Function test1

dlib/test/graph_labeler.cpp:290–331

        typename graph_type
        >
    void test1(
        const dlib::array<graph_type>& samples,
        const std::vector<std::vector<bool> >& labels
    )
    {
        dlog << LINFO << "begin test1()";

        structural_graph_labeling_trainer<vector_type> trainer;
        //trainer.be_verbose();
        trainer.set_epsilon(1e-12);
        graph_labeler<vector_type> labeler = trainer.train(samples, labels);


        // test serialization code for the labeler.
        std::ostringstream sout;
        serialize(labeler, sout);
        std::istringstream sin(sout.str());
        labeler = graph_labeler<vector_type>();
        deserialize(labeler, sin);

        std::vector<bool> temp;
        for (unsigned long k = 0; k < samples.size(); ++k)
        {
            temp = labeler(samples[k]);
            for (unsigned long i = 0; i < temp.size(); ++i)
            {
                const bool true_label = (labels[k][i] != 0);
                const bool pred_label = (temp[i] != 0);
                DLIB_TEST(true_label == pred_label);
            }
        }

        matrix<double> cv;

        cv = test_graph_labeling_function(labeler, samples, labels);
        DLIB_TEST(sum(cv) == 2);
        cv = cross_validate_graph_labeling_trainer(trainer, samples, labels, 4);
        DLIB_TEST(sum(cv) == 2);

        dlog << LINFO << "edge weights: " << trans(sparse_to_dense(labeler.get_edge_weights()));
        dlog << LINFO << "node weights: " << trans(sparse_to_dense(labeler.get_node_weights()));
    }