def test_discrete_predict(self):
train_data, test_data = load_project_data('example')
examples = [ex for ex in train_data] + [ex for ex in test_data]
data = ExampleSet(examples)
n = Node()
n.train(data, [1, 3]) #only train on the discrete data
self.assertTrue(all([ex[-1] == n.predict(ex) for ex in data]))
n.train(data, [3, 1]) #only train on the discrete data
self.assertTrue(all([ex[-1] == n.predict(ex) for ex in data]))
def test_discrete_predict(self):
train_data, test_data = load_project_data('example')
examples = [ex for ex in train_data] + [ex for ex in test_data]
data = ExampleSet(examples)
n = Node()
n.train(data,[1,3]) #only train on the discrete data
self.assertTrue(all([ex[-1]==n.predict(ex) for ex in data]))
n.train(data,[3,1]) #only train on the discrete data
self.assertTrue(all([ex[-1]==n.predict(ex) for ex in data]))
