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_starts_in_zero(self):
entropy = OnlineEntropy(lambda x: None)
self.assertEqual(entropy.get_entropy(), 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)
def test_starts_in_zero(self):
entropy = OnlineEntropy(lambda x: None)
self.assertEqual(entropy.get_entropy(), 0)
