Guest User

Untitled

a guest
Jun 1st, 2018
38
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.92 KB | None | 0 0
  1. TEST(Perceptron, custom_hyperplane_initialization)
  2. {
  3. auto env = linear_test_env->getBinaryLabelData();
  4. auto features = wrap(env->get_features_train());
  5. auto labels = wrap(env->get_labels_train());
  6. auto test_features = wrap(env->get_features_test());
  7. auto test_labels = wrap(env->get_labels_test());
  8.  
  9. auto perceptron = some<CPerceptron>(features, labels);
  10. perceptron->train();
  11.  
  12. auto weights = perceptron->get_w();
  13. weights.set_const(1.2);
  14. auto perceptron_initialized = some<CPerceptron>(features, labels);
  15. perceptron_initialized->set_initialize_hyperplane(false);
  16. perceptron_initialized->set_w(weights);
  17. perceptron_initialized->set_max_iter(1);
  18. SG_SPRINT("w: [0]=>%f [1]=>%f\n", weights[0], weights[1]);
  19. perceptron_initialized->train();
  20. SG_SPRINT("pw: [0]=>%f [1]=>%f\n", perceptron_initialized->get_w()[0], perceptron_initialized->get_w()[1]);
  21. EXPECT_TRUE(perceptron_initialized->get_w().equals(weights));
  22. }
Add Comment
Please, Sign In to add comment