Advertisement
Guest User

Untitled

a guest
Jun 20th, 2019
84
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.75 KB | None | 0 0
  1. from sklearn.neighbors import KNeighborsClassifier
  2. knn = KNeighborsClassifier()
  3. knn_model = knn.fit(X_train,y_train)
  4. #Use the values for your confusion matrix
  5. knn_y_model = knn_model.predict(X=X_test)
  6. # Use the probabilities for your ROC and Precision-recall curves
  7. knn_y_proba = knn_model.predict_proba(X=X_test)
  8.  
  9. from mlxtend.plotting import plot_confusion_matrix
  10. fig, ax = plot_confusion_matrix(conf_mat=confusion_matrix(y_test,knn_y_model),
  11. show_absolute=True,show_normed=True,colorbar=True)
  12. plt.title("Confusion matrix - KNN")
  13. plt.ylabel('True label')
  14. plt.xlabel('Predicted label'
  15.  
  16. import scikitplot as skplt
  17. plot = skplt.metrics.plot_roc(y_test, knn_y_proba)
  18. plt.title("ROC Curves - K-Nearest Neighbors")
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement