def start(self):
"""Get data from rabbitMQ and classify input data"""
if self._network is None:
self._network = Network(self.trained_network_path)
regionNames = self._get_all_regions_names()
setNetworkLearningMode(self._network, regionNames, False)
_LOGGER.info("[Module %s] Starting Motor Imagery module. Routing keys: %s"
% (self.module_id, self.routing_keys))
self.input_metric_subscriber.subscribe(
self.routing_keys[self.input_metric],
self._tag_and_classify)
_CONFIG = simplejson.load(open(_CONFIG_JSON, "rb"))
_REGION_CONFIG_KEYS = ("spRegionConfig", "tmRegionConfig",
"tpRegionConfig",
"classifierRegionConfig")
_REGION_NAMES = []
for region in _REGION_CONFIG_KEYS:
if _CONFIG[region].get("regionEnabled"):
_REGION_NAMES.append(_CONFIG[region]["regionName"])
if __name__ == "__main__":
dataSource = FileRecordStream(streamID="backup/training-data-attention.csv")
network = configureNetwork(dataSource, _CONFIG)
setNetworkLearningMode(network, _REGION_NAMES, True)
sensorRegion = network.regions[
_CONFIG["sensorRegionConfig"].get("regionName")]
classifierRegion = network.regions[
_CONFIG["classifierRegionConfig"].get("regionName")]
headers = ["x", "y", "label"]
num_correct = 0
for category in range(_NUM_CATEGORIES):
csvFile = open(_INPUT_FILES[category], "rb")
reader = csv.reader(csvFile)
# skip 3 header rows
reader.next()
reader.next()
