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