Skip to content

Commit 9bf40b5

Browse files
Updated
1 parent 4cbedf9 commit 9bf40b5

File tree

1 file changed

+10
-7
lines changed

1 file changed

+10
-7
lines changed

scikitplot/metrics.py

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,9 @@
22
The :mod:`scikitplot.metrics` module includes plots for machine learning
33
evaluation metrics e.g. confusion matrix, silhouette scores, etc.
44
"""
5-
from __future__ import absolute_import, division, print_function, \
6-
unicode_literals
5+
from __future__ import (
6+
absolute_import, division, print_function, unicode_literals
7+
)
78

89
import itertools
910

@@ -28,11 +29,13 @@
2829
from scikitplot.helpers import cumulative_gain_curve
2930

3031

31-
def plot_confusion_matrix(y_true, y_pred, labels=None, true_labels=None,
32-
pred_labels=None, title=None, normalize=False,
33-
hide_zeros=False, hide_counts=False, x_tick_rotation=0, ax=None,
34-
figsize=None, cmap='Blues', title_fontsize="large",
35-
text_fontsize="medium", show_colorbar=True):
32+
def plot_confusion_matrix(
33+
y_true, y_pred, labels=None, true_labels=None,
34+
pred_labels=None, title=None, normalize=False,
35+
hide_zeros=False, hide_counts=False, x_tick_rotation=0, ax=None,
36+
figsize=None, cmap='Blues', title_fontsize="large",
37+
text_fontsize="medium", show_colorbar=True,
38+
):
3639
"""Generates confusion matrix plot from predictions and true labels
3740
3841
Args:

0 commit comments

Comments
 (0)