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.initialize()
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.classify(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.initialize()