def testCSV(self): ml = ML() # ml.compute_statistics(iris, "sepallength") t = Training( "classification", "LogisticRegression", { "maxIter": 10, "regParam": 0.3, "elasticNetParam": 0.8, "family": "multinomial" }, 0.7, iris, "./model/lr") # ml.train_model(t) ml.model_predict_single(t, iris)
def testALS(self): ml = ML() t = Training("recommendation", "ALS", {"maxIter": 9}, 0.7, test, "./model/als") ml.model_predict_single() print(ml.train_model(t))