43405974e25fa600b38aa9a7e52a6e86d0bd0e59,yellowbrick/classifier/class_balance.py,ClassBalance,score,#ClassBalance#,71

Before Change

        y_pred = self.predict(X)
        self.scores  = precision_recall_fscore_support(y, y_pred)
        self.support = dict(zip(self.classes_, self.scores[-1]))
        return self.draw()

    def draw(self):
        
        Renders the class balance chart across the axis.

After Change

        self.scores  = precision_recall_fscore_support(y, y_pred)
        self.support = dict(zip(self.classes_, self.scores[-1]))

        self.draw()

        // Retrieve and store the score attribute from the sklearn classifier
        self.score_ = self.estimator.score(X, y)

        return self.score_

    def draw(self):
        
        Renders the class balance chart across the axis.

Instances