from dataapi import getDataSteam import sys import importlib import csv import random from Queue import Queue from nupic.frameworks.opf.modelfactory import ModelFactory from model_params import exercise_model_params predictionSteps = 1 model = ModelFactory.create(exercise_model_params.MODEL_PARAMS) model.enableInference({'predictedField': 'torso'}) resultFile = open("output.txt", 'w', 0) resultWriter = csv.writer(resultFile) resultWriter.writerow(["activity_person","torso","score"]) resultWriter.writerow(["string","float","float"]) resultWriter.writerow(["", "",""]) for activity, person in [(10, 5), (10, 6), (5, 6)]: activity_person = str(activity) + "_" + str(person) for line in getDataSteam(activity, person): torso = line[0] row = {"torso": torso} result = model.run(row) futurePrediction = result.inferences["multiStepBestPredictions"][predictionSteps] anomalyScore = result.inferences["anomalyScore"] resultWriter.writerow([activity_person, torso, anomalyScore])