Ejemplo n.º 1
0
def outputRegressions(parameters, path="regression_output"):
    os.makedirs(path, exist_ok=True)
    outputReport(path)
    outputDict(path, parameters)


def outputReport(path):
    pass


def outputDict(path, parameters):
    f = open(path+"/.regressions", "wb")
    pickle.dump(parameters, f)
    f.close()


if __name__ == '__main__':
    data = parseData() if len(sys.argv) == 1 else parseData(sys.argv[1])

    ac = AddressClassifier()
    ac.classify(data)
    rawData = ac.getClassifiedData()

    print ("regression_generator start")
    reg = Regression()
    reg.quantize(rawData)
    parameters = reg.getTotalParams()

    outputRegressions(parameters)
    print ("regression_generator finish")