PredictionKind

FastTreeClassification.cs (1)

182private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;

FastTreeRanking.cs (1)

82private protected override PredictionKind PredictionKind => PredictionKind.Ranking;

FastTreeRegression.cs (1)

73private protected override PredictionKind PredictionKind => PredictionKind.Regression;

FastTreeTweedie.cs (1)

75private protected override PredictionKind PredictionKind => PredictionKind.Regression;

GamClassification.cs (1)

79private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;

GamRegression.cs (1)

75private protected override PredictionKind PredictionKind => PredictionKind.Regression;

RandomForestClassification.cs (1)

175private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;

RandomForestRegression.cs (1)

310private protected override PredictionKind PredictionKind => PredictionKind.Regression;

KMeansPlusPlusTrainer.cs (1)

163private protected override PredictionKind PredictionKind => PredictionKind.Clustering;

LightGbmBinaryTrainer.cs (1)

123private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;

LightGbmMulticlassTrainer.cs (1)

69private protected override PredictionKind PredictionKind => PredictionKind.MulticlassClassification;

LightGbmRankingTrainer.cs (1)

112private protected override PredictionKind PredictionKind => PredictionKind.Ranking;

LightGbmRegressionTrainer.cs (1)

115private protected override PredictionKind PredictionKind => PredictionKind.Regression;

OlsLinearRegression.cs (1)

110private protected override PredictionKind PredictionKind => PredictionKind.Regression;

SymSgdClassificationTrainer.cs (1)

235private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;

PcaTrainer.cs (1)

127private protected override PredictionKind PredictionKind => PredictionKind.AnomalyDetection;

LdSvm\LdSvmTrainer.cs (1)

178private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;

Standard\LogisticRegression\LogisticRegression.cs (1)

161private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;

Standard\LogisticRegression\MulticlassLogisticRegression.cs (1)

167private protected override PredictionKind PredictionKind => PredictionKind.MulticlassClassification;

Standard\MulticlassClassification\MulticlassNaiveBayesTrainer.cs (1)

78private protected override PredictionKind PredictionKind => PredictionKind.MulticlassClassification;

Standard\Online\AveragedPerceptron.cs (1)

192private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;

Standard\Online\LinearSvm.cs (1)

303private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;

Standard\Online\OnlineGradientDescent.cs (1)

166private protected override PredictionKind PredictionKind => PredictionKind.Regression;

Standard\PoissonRegression\PoissonRegression.cs (1)

111private protected override PredictionKind PredictionKind => PredictionKind.Regression;

Standard\SdcaBinary.cs (2)

1457private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification; 1951private protected override PredictionKind PredictionKind => PredictionKind.BinaryClassification;

Standard\SdcaMulticlass.cs (1)

113private protected override PredictionKind PredictionKind => PredictionKind.MulticlassClassification;

Standard\SdcaRegression.cs (1)

98private protected override PredictionKind PredictionKind => PredictionKind.Regression;

ImageClassificationTrainer.cs (1)

460private protected override PredictionKind PredictionKind => PredictionKind.MulticlassClassification;

Training\TrainerEstimatorBase.cs (1)

52PredictionKind ITrainer.PredictionKind => PredictionKind;

FastTree.cs (3)

194TrainSet = instanceConverter.FindBinsAndReturnDataset(trainData, PredictionKind, ParallelTraining, CategoricalFeatures, FastTreeTrainerOptions.CategoricalSplit); 197ValidSet = instanceConverter.GetCompatibleDataset(ValidData, PredictionKind, CategoricalFeatures, FastTreeTrainerOptions.CategoricalSplit); 199TestSets = new[] { instanceConverter.GetCompatibleDataset(TestData, PredictionKind, CategoricalFeatures, FastTreeTrainerOptions.CategoricalSplit) };

GamTrainer.cs (2)

260TrainSet = instanceConverter.FindBinsAndReturnDataset(trainData, PredictionKind, ParallelTraining, null, false); 263ValidSet = instanceConverter.GetCompatibleDataset(validationData, PredictionKind, null, false);

LightGbmTrainerBase.cs (2)

554if (PredictionKind == PredictionKind.Ranking) 773if (PredictionKind == PredictionKind.Ranking)