def testClassifierMakerClassifiers(self): "Check that all classifiers that can be made are valid." data, test_data, vectors = testData() class_values = set(x[2] for x in data) params = {'beta': [1.0, 2.0], 'costs': [1.0], 'k': [1]} maker = ClassifierMaker(vectors, params) names = maker.get_names() for name in names: classifier = maker.make(name) classifier.fit(data) results = classifier.predict(test_data) self.assertEqual(len(results), len(test_data)) self.assertTrue(set(results) <= class_values)
def testClassifierMakerClassifiers(self): "Check that all classifiers that can be made are valid." data, test_data, vectors = testData() class_values = set(x[2] for x in data) params = {'beta':[1.0, 2.0], 'costs':[1.0]} maker = ClassifierMaker(vectors, params) names = maker.get_names() for name in names: classifier = maker.make(name) classifier.fit(data) results = classifier.predict(test_data) self.assertEqual(len(results), len(test_data)) self.assertTrue(set(results) <= class_values)
def testClassifierMakerNames(self): data, test_data, vectors = testData() maker = ClassifierMaker(vectors) names = maker.get_names() self.assertGreater(len(names), 0, "Maker should have more than one classifier")