Skip to content

Instantly share code, notes, and snippets.

@Davisy

Davisy/generate_comfusion_matrix.py

Created April 26, 2020 15:22
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save Davisy/b72173903483fae1a6b0389c478d6628 to your computer and use it in GitHub Desktop.
Save Davisy/b72173903483fae1a6b0389c478d6628 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
generate_comfusion_matrix.py
y_pred = xgb_classifier.predict(X_val)
# plot confusion matrix for xgb_classifier
skplt.metrics.plot_confusion_matrix(
y_val,
y_pred,
normalize=True,
title="Confusion Matrix for XGB Classifier",
figsize=(10, 8),
)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment