5 implementations of PredictionKind
Microsoft.ML.Data (3)
Dirty\ModelParametersBase.cs (1)
69PredictionKind IPredictor.PredictionKind => PredictionKind;
Prediction\Calibrator.cs (1)
209PredictionKind IPredictor.PredictionKind => ((IPredictorProducing<float>)SubModel).PredictionKind;
Scorers\FeatureContributionCalculation.cs (1)
117public PredictionKind PredictionKind => Predictor.PredictionKind;
Microsoft.ML.Ensemble (1)
PipelineEnsemble.cs (1)
399public abstract PredictionKind PredictionKind { get; }
Microsoft.ML.Recommender (1)
MatrixFactorizationPredictor.cs (1)
85PredictionKind IPredictor.PredictionKind => PredictionKind.Recommendation;
23 references to PredictionKind
Microsoft.ML.Data (13)
DataLoadSave\TransformerChain.cs (7)
300if (predictor.PredictionKind == PredictionKind.BinaryClassification) 303else if (predictor.PredictionKind == PredictionKind.MulticlassClassification) 308else if (predictor.PredictionKind == PredictionKind.Clustering) 311else if (predictor.PredictionKind == PredictionKind.Regression) 314else if (predictor.PredictionKind == PredictionKind.AnomalyDetection) 317else if (predictor.PredictionKind == PredictionKind.Ranking) 321throw env.Except("Don't know how to map prediction kind {0}", predictor.PredictionKind);
Prediction\Calibrator.cs (1)
209PredictionKind IPredictor.PredictionKind => ((IPredictorProducing<float>)SubModel).PredictionKind;
Scorers\FeatureContributionCalculation.cs (1)
117public PredictionKind PredictionKind => Predictor.PredictionKind;
Scorers\SchemaBindablePredictorWrapper.cs (4)
362switch (predictor.PredictionKind) 379throw Contracts.Except("Unknown prediction kind, can't map to score column kind: {0}", predictor.PredictionKind); 494if (Predictor.PredictionKind != PredictionKind.BinaryClassification) 495ch.Warning("Scoring predictor of kind '{0}' as '{1}'.", Predictor.PredictionKind, PredictionKind.BinaryClassification);
Microsoft.ML.Ensemble (3)
Trainer\Binary\EnsembleTrainer.cs (2)
99return new EnsembleDistributionModelParameters(Host, p.PredictionKind, 104return new EnsembleModelParameters(Host, p.PredictionKind,
Trainer\Regression\RegressionEnsembleTrainer.cs (1)
91var predictor = new EnsembleModelParameters(Host, p.PredictionKind,
Microsoft.ML.EntryPoints (5)
PermutationFeatureImportance.cs (4)
68if (predictor.PredictionKind == PredictionKind.BinaryClassification) 70else if (predictor.PredictionKind == PredictionKind.MulticlassClassification) 72else if (predictor.PredictionKind == PredictionKind.Regression) 74else if (predictor.PredictionKind == PredictionKind.Ranking)
ScoreColumnSelector.cs (1)
73if (input.PredictorModel.Predictor.PredictionKind == PredictionKind.BinaryClassification)
Microsoft.ML.OnnxConverter (2)
SaveOnnxCommand.cs (2)
359if (rawPred.PredictionKind == PredictionKind.BinaryClassification || rawPred.PredictionKind == PredictionKind.MulticlassClassification)