/* Regression */ func RegAlgorithmRun(regressor algo.Regressor, train_path string, test_path string, pred_path string, params map[string]string) (float64, []*eval.RealPrediction, error) { global, _ := strconv.ParseInt(params["global"], 10, 64) train_dataset := core.NewRealDataSet() err := train_dataset.Load(train_path, global) if err != nil { return 0.5, nil, err } test_dataset := core.NewRealDataSet() err = test_dataset.Load(test_path, global) if err != nil { return 0.5, nil, err } regressor.Init(params) rmse, predictions := RegAlgorithmRunOnDataSet(regressor, train_dataset, test_dataset, pred_path, params) return rmse, predictions, nil }
func RegAlgorithmTest(regressor algo.Regressor, test_path string, pred_path string, params map[string]string) (float64, []*eval.RealPrediction, error) { global, _ := strconv.ParseInt(params["global"], 10, 64) model_path, _ := params["model"] regressor.Init(params) if model_path != "" { regressor.LoadModel(model_path) } else { return 0.0, nil, nil } test_dataset := core.NewRealDataSet() err := test_dataset.Load(test_path, global) if err != nil { return 0.0, nil, err } rmse, predictions := RegAlgorithmRunOnDataSet(regressor, nil, test_dataset, pred_path, params) return rmse, predictions, nil }
func RegAlgorithmTrain(regressor algo.Regressor, train_path string, params map[string]string) error { global, _ := strconv.ParseInt(params["global"], 10, 64) train_dataset := core.NewRealDataSet() err := train_dataset.Load(train_path, global) if err != nil { return err } regressor.Init(params) regressor.Train(train_dataset) model_path, _ := params["model"] if model_path != "" { regressor.SaveModel(model_path) } return nil }