From d57102cd07b51781e9ff8b382268c2bc94bad2e6 Mon Sep 17 00:00:00 2001
From: Rebecca Bilbro <rebecca.bilbro@bytecubed.com>
Date: Sat, 8 Oct 2016 07:01:24 -0400
Subject: [PATCH] changes self.classes to self.classes_ to mimic Scikit-learn

---
 yellowbrick/classifier.py | 24 ++++++++++++------------
 1 file changed, 12 insertions(+), 12 deletions(-)

diff --git a/yellowbrick/classifier.py b/yellowbrick/classifier.py
index 4974b2d20..b45b476a2 100644
--- a/yellowbrick/classifier.py
+++ b/yellowbrick/classifier.py
@@ -67,7 +67,7 @@ def __init__(self, model, **kwargs):
         self.estimator = model
         self.name = get_model_name(self.estimator)
         self.cmap = kwargs.pop('cmap', ddlheatmap)
-        self.classes = model.classes_
+        self.classes_ = model.classes_
 
 
     def score(self, y, y_pred=None, **kwargs):
@@ -75,8 +75,8 @@ def score(self, y, y_pred=None, **kwargs):
         Generates the Scikit-Learn classification_report
         """
         self.keys = ('precision', 'recall', 'f1')
-        self.scores = precision_recall_fscore_support(y, y_pred, labels=self.classes)
-        self.scores = map(lambda s: dict(zip(self.classes, s)), self.scores[0:3])
+        self.scores = precision_recall_fscore_support(y, y_pred, labels=self.classes_)
+        self.scores = map(lambda s: dict(zip(self.classes_, s)), self.scores[0:3])
         self.scores = dict(zip(self.keys, self.scores))
         self.draw(y, y_pred)
 
@@ -88,11 +88,11 @@ def draw(self, y, y_pred):
         fig, ax = plt.subplots(1)
 
         self.matrix = []
-        for cls in self.classes:
+        for cls in self.classes_:
             self.matrix.append([self.scores['precision'][cls],self.scores['recall'][cls],self.scores['f1'][cls]])
 
         for column in range(len(self.matrix)+1):
-            for row in range(len(self.classes)):
+            for row in range(len(self.classes_)):
                 ax.text(column,row,self.matrix[row][column],va='center',ha='center')
 
         fig = plt.imshow(self.matrix, interpolation='nearest', cmap=self.cmap)
@@ -105,10 +105,10 @@ def poof(self):
         """
         plt.title('{} Classification Report'.format(self.name))
         plt.colorbar()
-        x_tick_marks = np.arange(len(self.classes)+1)
-        y_tick_marks = np.arange(len(self.classes))
+        x_tick_marks = np.arange(len(self.classes_)+1)
+        y_tick_marks = np.arange(len(self.classes_))
         plt.xticks(x_tick_marks, ['precision', 'recall', 'f1-score'], rotation=45)
-        plt.yticks(y_tick_marks, self.classes)
+        plt.yticks(y_tick_marks, self.classes_)
         plt.ylabel('Classes')
         plt.xlabel('Measures')
 
@@ -195,14 +195,14 @@ def __init__(self, model, **kwargs):
         self.estimator = model
         self.name      = get_model_name(self.estimator)
         self.colors    = kwargs.pop('colors', YELLOWBRICK_PALETTES['paired'])
-        self.classes   = model.classes_
+        self.classes_   = model.classes_
 
     def score(self, y, y_pred=None, **kwargs):
         """
         Generates the Scikit-Learn precision_recall_fscore_support
         """
-        self.scores  = precision_recall_fscore_support(y, y_pred, labels=self.classes)
-        self.support = dict(zip(self.classes, self.scores[-1])  )
+        self.scores  = precision_recall_fscore_support(y, y_pred, labels=self.classes_)
+        self.support = dict(zip(self.classes_, self.scores[-1])  )
         self.draw()
 
     def draw(self):
@@ -213,7 +213,7 @@ def draw(self):
         Refactor to make better use of yb_palettes module?
 
         """
-        colors = self.colors[0:len(self.classes)]
+        colors = self.colors[0:len(self.classes_)]
         plt.bar(np.arange(len(self.support)), self.support.values(), color=colors, align='center', width=0.5)
 
     def poof(self):