diff --git a/docs/conf.py b/docs/conf.py index b57637c..a6fdad9 100644 --- a/docs/conf.py +++ b/docs/conf.py @@ -48,7 +48,7 @@ # The short X.Y version version = '0.12' # The full version, including alpha/beta/rc tags -release = '0.12.3' +release = '0.12.4' # -- General configuration --------------------------------------------------- diff --git a/neptunecontrib/monitoring/metrics.py b/neptunecontrib/monitoring/metrics.py index 577b1c0..27afbef 100644 --- a/neptunecontrib/monitoring/metrics.py +++ b/neptunecontrib/monitoring/metrics.py @@ -694,9 +694,13 @@ def expand_prediction(prediction): def _plot_confusion_matrix(y_true, y_pred_class, ax=None): + if not ax: + _, ax = plt.subplots() cmap = plt.get_cmap('Blues') cm = sk_metrics.confusion_matrix(y_true, y_pred_class) sns.heatmap(cm, cmap=cmap, annot=True, fmt='g', ax=ax) + ax.set_xlabel('predicted values') + ax.set_ylabel('actual values') def _plot_class_metrics_by_threshold(y_true, y_pred_positive): @@ -714,6 +718,8 @@ def _plot_class_metrics_by_threshold(y_true, y_pred_positive): ax.text(x=best_thres + 0.01, y=0.98 * best_score, s='thres={:.4f}\nscore={:.4f}'.format(best_thres, best_score), color='red') + ax.set_xlabel('threshold') + ax.set_ylabel(name) figs.append(fig) return figs diff --git a/setup.py b/setup.py index 17ebbe0..0ab3e97 100644 --- a/setup.py +++ b/setup.py @@ -20,7 +20,7 @@ def main(): setup( name='neptune-contrib', - version='0.12.3', + version='0.12.4', description='Neptune Python library contributions', author='neptune.ml', author_email='contact@neptune.ml',