def test_classless(self): classless = Table(Domain(self.zoo.domain.attributes), self.zoo[:, 0:-1]) scorers = [score.Gini(), score.InfoGain(), score.GainRatio()] for scorer in scorers: with self.assertRaises(ValueError): scorer(0, classless)
def test_wrong_class_type(self): scorers = [score.Gini(), score.InfoGain(), score.GainRatio()] for scorer in scorers: with self.assertRaises(ValueError): scorer(0, self.housing) with self.assertRaises(ValueError): score.Chi2(2, self.housing) with self.assertRaises(ValueError): score.ANOVA(2, self.housing) score.UnivariateLinearRegression(2, self.housing)
def test_info_gain(self): scorer = score.InfoGain() correct = [0.79067, 0.71795, 0.83014, 0.97432, 0.46970] np.testing.assert_almost_equal([scorer(a, self.zoo) for a in range(5)], correct, decimal=5)