plot_results
Plot of the model results after the evaluation. If all models applied
bootstrap, the plot is a boxplot. If not, the plot is a barplot. Models
are ordered based on their score from the top down. The score is either
the mean_bootstrap
or metric_test
attribute of the model, selected in
that order.
Parameters: |
models: str, sequence or None, optional (default=None)
metric: int or str, optional (default=0)
title: str or None, optional (default=None)
figsize: tuple, optional (default=None)
filename: str or None, optional (default=None)
display: bool or None, optional (default=True) |
Returns: |
fig: matplotlib.figure.Figure Plot object. Only returned if display=None .
|
Example
from atom import ATOMClassifier
atom = ATOMClassifier(X, y)
atom.run(["QDA", "Tree", "RF", "ET", "LGB"], metric="f1", n_bootstrap=5)
atom.plot_results() # With bootstrap...

# And without bootstrap...
atom.run(["QDA", "Tree", "RF", "ET", "LGB"], metric="f1", n_bootstrap=0)
atom.plot_results()
