def test_color_sequence_bounds(self): """ Test color sequence out of bounds value error """ with self.assertRaises(YellowbrickValueError): color_sequence('RdBu', 18) with self.assertRaises(YellowbrickValueError): color_sequence('RdBu', 2)
def test_color_sequence_bounds(self): """ Test color sequence out of bounds value error """ with self.assertRaises(YellowbrickValueError): cmap = color_sequence('RdBu', 18) with self.assertRaises(YellowbrickValueError): cmap = color_sequence('RdBu', 2)
def test_color_sequence_bounds(self): """ Test color sequence out of bounds value error """ with pytest.raises(YellowbrickValueError): color_sequence("RdBu", 18) with pytest.raises(YellowbrickValueError): color_sequence("RdBu", 2)
def __init__(self, model, ax=None, sample_weight=None, percent=False, classes=None, encoder=None, cmap="YlOrRd", fontsize=None, is_fitted="auto", force_model=False, **kwargs): super(ConfusionMatrix, self).__init__(model, ax=ax, classes=classes, encoder=encoder, is_fitted=is_fitted, force_model=force_model, **kwargs) # Visual parameters self.fontsize = fontsize self.cmap = color_sequence(cmap) self.cmap.set_under(color=CMAP_UNDERCOLOR) self.cmap.set_over(color=CMAP_OVERCOLOR) # Estimator parameters self.percent = percent self.sample_weight = sample_weight # Used to draw diagonal line for predicted class = true class self._edgecolors = []
def test_color_sequence_default(self): """ Assert the default color sequence is RdBu """ cmap = color_sequence() self.assertEqual(cmap.name, "RdBu") self.assertEqual(cmap.N, 11)
def __init__(self, estimator, ax=None, classes=None, cmap="YlOrRd", support=None, encoder=None, is_fitted="auto", force_model=False, **kwargs): super(ClassificationReport, self).__init__(estimator, ax=ax, classes=classes, encoder=encoder, is_fitted=is_fitted, force_model=force_model, **kwargs) self.support = support self.cmap = color_sequence(cmap) self.cmap.set_over(color=CMAP_OVERCOLOR) self.cmap.set_under(color=CMAP_UNDERCOLOR) self._displayed_scores = [key for key in SCORES_KEYS] if support not in {None, True, False, "percent", "count"}: raise YellowbrickValueError( "'{}' is an invalid argument for support, use None, True, " "False, 'percent', or 'count'".format(support)) if not support: self._displayed_scores.remove("support")
def test_color_sequence_default(self): """ Assert the default color sequence is RdBu """ cmap = color_sequence() assert cmap.name == "RdBu" assert cmap.N == 11
def test_color_sequence(self): """ Ensure the color sequence returns listed colors. """ for name, ncols in SEQUENCES.items(): for n in ncols.keys(): cmap = color_sequence(name, n) self.assertEqual(name, cmap.name) self.assertEqual(n, cmap.N)
def test_color_sequence(self): """ Ensure the color sequence returns listed colors. """ for name, ncols in SEQUENCES.items(): for n in ncols.keys(): cmap = color_sequence(name, n) assert name == cmap.name assert n == cmap.N
def __init__(self, model, test_data, y_true, classes, name, path_to_save="./"): self.model = model self.test_data = test_data y_pred = model.predict(test_data) if len(y_true.shape) > 1: self.y_true = np.argmax(y_true, axis=1) self.y_pred = np.argmax(y_pred, axis=1) else: self.y_true = y_true self.y_pred = y_pred self.classes = classes self.name = name self.path_to_save = path_to_save self.cmap = color_sequence("YlOrRd") self.cmap.set_over(color="w") self.cmap.set_under(color="#2a7d4f") self._edgecolors = [] self.fontsize = None
def test_color_sequence_unrecocognized(self): """ Test value errors for unrecognized sequences """ with self.assertRaises(YellowbrickValueError): color_sequence('PepperBucks', 3)
def test_color_sequence_unrecocognized(self): """ Test value errors for unrecognized sequences """ with self.assertRaises(YellowbrickValueError): cmap = color_sequence('PepperBucks', 3)
def test_color_sequence_unrecocognized(self): """ Test value errors for unrecognized sequences """ with pytest.raises(YellowbrickValueError): color_sequence("PepperBucks", 3)