class HTMClassifierTest(unittest.TestCase):
def setUp(self):
with open(_NTWK_CONFIG, "rb") as jsonFile:
self.network_config = simplejson.load(jsonFile)
self.classifier = HTMClassifier(self.network_config, _TRAINING_DATA, _CATEGORIES)
self.classifier.connect()
def testTrainingAccuracy(self):
partitions = generateNetworkPartitions(self.network_config, _TRAIN_SET_SIZE)
training_accuracy = self.classifier.train(_TRAIN_SET_SIZE, partitions)
self.assertEqual(training_accuracy, 94.75, "classification accuracy is incorrect")
def testClassificationAccuracy(self):
mu = 7
tag = "middle"
result = self.classifier.start(input_data=mu, target=tag, learning_is_on=True)
print "classification_result: %s" % result
def setUp(self):
with open(_NTWK_CONFIG, "rb") as jsonFile:
self.network_config = simplejson.load(jsonFile)
self.classifier = HTMClassifier(self.network_config, _TRAINING_DATA, _CATEGORIES)
self.classifier.connect()
