public override void SetAlgorithm(object algo)
{
this.TrainingAlgo = (RegressionTrainingAlgorithm)algo;
}
public EuroPredictor()
{
_metrics = MetricsInitializer.Metrics;
trainTestDataRate = 0.2;
TrainingAlgo = RegressionTrainingAlgorithm.FastTree;
}
/*
* Setting Regression Training Algorithms
*/
public static IEstimator <ITransformer> ApplyRegressionTrainingAlgorithm(MLContext mlContext, RegressionTrainingAlgorithm trainingAlgorithm)
{
switch (trainingAlgorithm)
{
case RegressionTrainingAlgorithm.FastTree:
return(mlContext.Regression.Trainers.FastTree(labelColumnName: "Label", featureColumnName: "Features"));
case RegressionTrainingAlgorithm.FastTreeTweedie:
return(mlContext.Regression.Trainers.FastTreeTweedie(labelColumnName: "Label", featureColumnName: "Features"));
case RegressionTrainingAlgorithm.Gam:
return(mlContext.Regression.Trainers.Gam(labelColumnName: "Label", featureColumnName: "Features"));
default:
return(mlContext.Regression.Trainers.OnlineGradientDescent(labelColumnName: "Label", featureColumnName: "Features"));
}
}
