Function test1

dlib/test/assignment_learning.cpp:283–349

    template <typename fe_type, typename F>
    void test1(F make_data, bool force_assignment)
    {
        print_spinner();

        std::vector<sample_type> samples;
        std::vector<label_type> labels;

        make_data(samples, labels);
        make_data(samples, labels);
        make_data(samples, labels);

        randomize_samples(samples, labels);

        structural_assignment_trainer<fe_type> trainer;

        DLIB_TEST(trainer.forces_assignment() == false);
        DLIB_TEST(trainer.get_c() == 100);
        DLIB_TEST(trainer.get_num_threads() == 2);
        DLIB_TEST(trainer.get_max_cache_size() == 5);


        trainer.set_forces_assignment(force_assignment);
        trainer.set_num_threads(3);
        trainer.set_c(50);

        DLIB_TEST(trainer.get_c() == 50);
        DLIB_TEST(trainer.get_num_threads() == 3);
        DLIB_TEST(trainer.forces_assignment() == force_assignment);

        assignment_function<fe_type> ass = trainer.train(samples, labels);

        for (unsigned long i = 0; i < samples.size(); ++i)
        {
            std::vector<long> out = ass(samples[i]);
            dlog << LINFO << "true labels: " << trans(mat(labels[i]));
            dlog << LINFO << "pred labels: " << trans(mat(out));
            DLIB_TEST(trans(mat(labels[i])) == trans(mat(out)));
        }

        double accuracy;

        dlog << LINFO << "samples.size(): "<< samples.size();
        accuracy = test_assignment_function(ass, samples, labels);
        dlog << LINFO << "accuracy: "<< accuracy;
        DLIB_TEST(accuracy == 1);

        accuracy = cross_validate_assignment_trainer(trainer, samples, labels, 3);
        dlog << LINFO << "cv accuracy: "<< accuracy;
        DLIB_TEST(accuracy == 1);

        ostringstream sout;
        serialize(ass, sout);
        istringstream sin(sout.str());
        assignment_function<fe_type> ass2;
        deserialize(ass2, sin);

        DLIB_TEST(ass2.forces_assignment() == ass.forces_assignment());
        DLIB_TEST(length(ass2.get_weights() - ass.get_weights()) < 1e-10);