Skip to content

Instantly share code, notes, and snippets.

@pycaret

pycaret/plot_model.py

Created August 1, 2020 01:19
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 pycaret/87d46b76b3f4a05c51323377fd1d40aa to your computer and use it in GitHub Desktop.
Save pycaret/87d46b76b3f4a05c51323377fd1d40aa to your computer and use it in GitHub Desktop.
Download ZIP
Raw
plot_model.py
# import classification module
from pycaret.classification import *
# init setup
clf1 = setup(data, target = 'name-of-target')
# train adaboost model
adaboost = create_model('ada')
# AUC plot
plot_model(adaboost, plot = 'auc')
# Decision Boundary
plot_model(adaboost, plot = 'boundary')
# Precision Recall Curve
plot_model(adaboost, plot = 'pr')
# Validation Curve
plot_model(adaboost, plot = 'vc')
@gh-ga
Copy link

gh-ga commented Feb 6, 2023

How can you change the color of the lines in the plot? It automatically chooses very ambiguous colors for AUC at least (the micro-average lines have identical styles)
Screenshot 2023-02-06 133309

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment