Advertisement
toweber

knn_aula_p1

Sep 17th, 2021
242
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.34 KB | None | 0 0
  1. from sklearn import neighbors
  2.  
  3. X = [ [2.0,1.5,1.0], [1.0,1.0,0.5], [1.0,3.0,1.5], [0.8,2.2,0.7], [3.0,1.0,0.2], [0.4,2.0,1.7], [1.2,1.2,1.5]  ]
  4. Y = [ 1, 0, 1, 0, 1, 1, 1 ]
  5.  
  6.  
  7. clf = neighbors.KNeighborsClassifier(n_neighbors=3) #k =1, 3, 5, 7
  8. clf.fit(X, Y)
  9. print(clf.predict([[0.9, 1.1, 0.5]]))
  10. print(clf.predict_proba([[0.9, 1.1, 0.5]]))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement