Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- TEST(Perceptron, custom_hyperplane_initialization)
- {
- auto env = linear_test_env->getBinaryLabelData();
- auto features = wrap(env->get_features_train());
- auto labels = wrap(env->get_labels_train());
- auto test_features = wrap(env->get_features_test());
- auto test_labels = wrap(env->get_labels_test());
- auto perceptron = some<CPerceptron>(features, labels);
- perceptron->train();
- auto weights = perceptron->get_w();
- weights.set_const(1.2);
- auto perceptron_initialized = some<CPerceptron>(features, labels);
- perceptron_initialized->set_initialize_hyperplane(false);
- perceptron_initialized->set_w(weights);
- perceptron_initialized->set_max_iter(1);
- SG_SPRINT("w: [0]=>%f [1]=>%f\n", weights[0], weights[1]);
- perceptron_initialized->train();
- SG_SPRINT("pw: [0]=>%f [1]=>%f\n", perceptron_initialized->get_w()[0], perceptron_initialized->get_w()[1]);
- EXPECT_TRUE(perceptron_initialized->get_w().equals(weights));
- }
Add Comment
Please, Sign In to add comment