scikitplot.estimators#

This package/module is designed to be compatible with both Python 2 and Python 3. The imports below ensure consistent behavior across different Python versions by enforcing Python 3-like behavior in Python 2.

The estimators module includes plots built specifically for scikit-learn estimator (classifier/regressor) instances e.g. Random Forest. You can use your own estimators, but these plots assume specific properties shared by scikit-learn estimators. The specific requirements are documented per function.

User guide. See the Estimators section for further details.

plot_feature_importances

Generates a plot of a sklearn model's feature importances.

plot_learning_curve

Generates a plot of the train and test learning curves for a classifier.