def setUp(self):
self.feature_dict = {
'a': {'c1': 2, 'c2': 0},
'b': {'c1': 1, 'c2': 0},
'c': {'c1': 1, 'c2': 1},
'd': {'c1': 1, 'c2': 2}}
self.cate_dict = {'c1': 2, 'c2': 2}
self.weighter = InformationGain(self.feature_dict, self.cate_dict)
def setUp(self):
self.feature_dict = {
'a': {
'c1': 2,
'c2': 0
},
'b': {
'c1': 1,
'c2': 0
},
'c': {
'c1': 1,
'c2': 1
},
'd': {
'c1': 1,
'c2': 2
}
}
self.cate_dict = {'c1': 2, 'c2': 2}
self.weighter = InformationGain(self.feature_dict, self.cate_dict)
class TestInformationGain(unittest.TestCase):
def setUp(self):
self.feature_dict = {
'a': {
'c1': 2,
'c2': 0
},
'b': {
'c1': 1,
'c2': 0
},
'c': {
'c1': 1,
'c2': 1
},
'd': {
'c1': 1,
'c2': 2
}
}
self.cate_dict = {'c1': 2, 'c2': 2}
self.weighter = InformationGain(self.feature_dict, self.cate_dict)
def test_get_weight(self):
self.assertEquals(self.weighter.get_weight('a'), 1.)
self.assertEquals(self.weighter.get_weight('b'),
float('0.31127812445913283'))
self.assertEquals(self.weighter.get_weight('c'), 0.)
self.assertEquals(self.weighter.get_weight('d'),
float('0.31127812445913283'))
def test_performance(self):
start = time()
for x in xrange(100000):
self.weighter.get_weight('a')
print 'costs: %0.2f secs' % (time() - start)
class TestInformationGain(unittest.TestCase):
def setUp(self):
self.feature_dict = {
'a': {'c1': 2, 'c2': 0},
'b': {'c1': 1, 'c2': 0},
'c': {'c1': 1, 'c2': 1},
'd': {'c1': 1, 'c2': 2}}
self.cate_dict = {'c1': 2, 'c2': 2}
self.weighter = InformationGain(self.feature_dict, self.cate_dict)
def test_get_weight(self):
self.assertEquals(self.weighter.get_weight('a'), 1.)
self.assertEquals(
self.weighter.get_weight('b'), float('0.31127812445913283'))
self.assertEquals(self.weighter.get_weight('c'), 0.)
self.assertEquals(
self.weighter.get_weight('d'), float('0.31127812445913283'))
def test_performance(self):
start = time()
for x in xrange(100000):
self.weighter.get_weight('a')
print 'costs: %0.2f secs' % (time() - start)
