Advertisement
Guest User

Untitled

a guest
Jul 18th, 2019
81
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.57 KB | None | 0 0
  1. #include <iostream>
  2. #include "Dataset.h"
  3. #include "Knn.h"
  4. int main() {
  5.  
  6. Dataset randomized_data(100, 3);
  7. randomized_data.Print();
  8.  
  9. //Create a 2d vector a get prediction
  10. std::vector<float> value;
  11. value.push_back(.3f);
  12. value.push_back(.8f);
  13.  
  14.  
  15.  
  16. Knn knn = Knn();
  17. knn.Fit(randomized_data);
  18. int prediction=knn.Predict(value,7);
  19.  
  20. for (size_t i = 0; i < knn.Neighbors.size(); i++)
  21. {
  22. std::cout<<knn.Neighbors.at(i).first.at(0)<<" " << knn.Neighbors.at(i).first.at(1) <<std::endl;
  23. std::cout << knn.Neighbors.at(i).second << std::endl;
  24. }
  25. std::cout << "Labl:"<<prediction;
  26.  
  27.  
  28.  
  29. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement