def test_full_gain(self): target = lambda x: x % 7 gain = OnlineInformationGain(target, target) entropy = OnlineEntropy(target) for i in range(50): gain.add(i) entropy.add(i) self.assertEqual(gain.get_gain(), entropy.get_entropy()) self.assertGreaterEqual(gain.get_gain(), 0)
def test_full_gain(self): target = lambda x: x % 7 gain = OnlineInformationGain(target, target) entropy = OnlineEntropy(target) for i in xrange(50): gain.add(i) entropy.add(i) self.assertEqual(gain.get_gain(), entropy.get_entropy()) self.assertGreaterEqual(gain.get_gain(), 0)
def test_valid_values(self): entropy = OnlineEntropy(lambda x: x % 10) for i in range(150): entropy.add(i) self.assertGreaterEqual(entropy.get_entropy(), 0.0)
def test_valid_values(self): entropy = OnlineEntropy(lambda x: x % 10) for i in xrange(150): entropy.add(i) self.assertGreaterEqual(entropy.get_entropy(), 0.0)