public BankNotePredictor() { _metrics = MetricsInitializer.Metrics; this.trainTestDataRate = 0.2; TrainingAlgorithm = BinaryClassificationTrainingAlgorithm.LbfgsLogisticRegression; }
/* * Setting Binary Classification Training Algorithms */ public static IEstimator <ITransformer> ApplyBinaryTrainingAlgorithm(MLContext mlContext, BinaryClassificationTrainingAlgorithm trainingAlgorithm) { switch (trainingAlgorithm) { case BinaryClassificationTrainingAlgorithm.FastTree: return(mlContext.BinaryClassification.Trainers.FastTree()); case BinaryClassificationTrainingAlgorithm.LbfgsLogisticRegression: return(mlContext.BinaryClassification.Trainers.LbfgsLogisticRegression()); case BinaryClassificationTrainingAlgorithm.FieldAwareFactorizationMachine: return(mlContext.BinaryClassification.Trainers.FieldAwareFactorizationMachine()); case BinaryClassificationTrainingAlgorithm.Gam: return(mlContext.BinaryClassification.Trainers.Gam()); default: return(mlContext.BinaryClassification.Trainers.FastTree()); } }
/* * Setting training algorithm */ public override void SetAlgorithm(object algo) { this.TrainingAlgorithm = (BinaryClassificationTrainingAlgorithm)algo; }