示例#1
0
 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)
示例#2
0
 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))