18 references to TDistPredictor
Microsoft.ML.Ensemble (5)
Trainer\Binary\EnsembleTrainer.cs (5)
85if (models.All(m => m.Predictor is TDistPredictor)) 86return new EnsembleDistributionModelParameters(Host, PredictionKind, CreateModels<TDistPredictor>(models), Combiner); 96if (p is TDistPredictor) 98Host.CheckParam(models.All(m => m is TDistPredictor), nameof(models)); 100models.Select(k => new FeatureSubsetModel<float>((TDistPredictor)k)).ToArray(), combiner);
Microsoft.ML.StandardTrainers (13)
Standard\MulticlassClassification\OneVersusAllTrainer.cs (2)
175var calibratedModel = transformer.Model as TDistPredictor; 182calibratedModel = CalibratorUtils.GetCalibratedPredictor(Host, ch, Calibrator, transformer.Model, trainedData, Args.MaxCalibrationExamples) as TDistPredictor;
Standard\MulticlassClassification\PairwiseCouplingTrainer.cs (11)
129var predModels = new TDistPredictor[count][]; 133predModels[i] = new TDistPredictor[i + 1]; 157var calibratedModel = transformer.Model as TDistPredictor; 159calibratedModel = CalibratorUtils.GetCalibratedPredictor(Host, ch, Calibrator, transformer.Model, trainedData, Args.MaxCalibrationExamples) as TDistPredictor; 161return new BinaryPredictionTransformer<TDistPredictor>(Host, calibratedModel, trainedData.Data.Schema, transformer.FeatureColumnName); 196var predictors = new TDistPredictor[numClasses][]; 202predictors[i] = new TDistPredictor[i + 1]; 269_predictors = new TDistPredictor[checked(_numClasses * (_numClasses + 1) / 2)]; 298_predictors = new TDistPredictor[(int)count]; 305ctx.LoadModel<TDistPredictor, SignatureLoadModel>(Host, out _predictors[index++], string.Format(SubPredictorFmt2, i, j)); 308ctx.LoadModel<TDistPredictor, SignatureLoadModel>(Host, out _predictors[index++], string.Format(SubPredictorFmt, i));