4 instantiations of FastTreeBinaryModelParameters
Microsoft.ML.FastTree (4)
FastTreeClassification.cs (2)
92var predictor = new FastTreeBinaryModelParameters(env, ctx); 208var pred = new FastTreeBinaryModelParameters(Host, TrainedEnsemble, FeatureCount, InnerOptions);
TreeEnsemble\TreeEnsembleCombiner.cs (2)
104return new FastTreeBinaryModelParameters(_host, ensemble, featureCount, null); 107var fastTreeModel = new FastTreeBinaryModelParameters(_host, ensemble, featureCount, null);
30 references to FastTreeBinaryModelParameters
Microsoft.ML.FastTree (17)
FastTreeClassification.cs (14)
35[assembly: LoadableClass(typeof(IPredictorProducing<float>), typeof(FastTreeBinaryModelParameters), null, typeof(SignatureLoadModel), 37FastTreeBinaryModelParameters.LoaderSignature)] 62loaderAssemblyName: typeof(FastTreeBinaryModelParameters).Assembly.FullName); 92var predictor = new FastTreeBinaryModelParameters(env, ctx); 99return new SchemaBindableCalibratedModelParameters<FastTreeBinaryModelParameters, ICalibrator>(env, predictor, calibrator); 133BinaryPredictionTransformer<CalibratedModelParametersBase<FastTreeBinaryModelParameters, PlattCalibrator>>, 134CalibratedModelParametersBase<FastTreeBinaryModelParameters, PlattCalibrator>> 186private protected override CalibratedModelParametersBase<FastTreeBinaryModelParameters, PlattCalibrator> TrainModelCore(TrainContext context) 208var pred = new FastTreeBinaryModelParameters(Host, TrainedEnsemble, FeatureCount, InnerOptions); 215return new FeatureWeightsCalibratedModelParameters<FastTreeBinaryModelParameters, PlattCalibrator>(Host, pred, cali); 303private protected override BinaryPredictionTransformer<CalibratedModelParametersBase<FastTreeBinaryModelParameters, PlattCalibrator>> MakeTransformer( 304CalibratedModelParametersBase<FastTreeBinaryModelParameters, PlattCalibrator> model, DataViewSchema trainSchema) 305=> new BinaryPredictionTransformer<CalibratedModelParametersBase<FastTreeBinaryModelParameters, PlattCalibrator>>(Host, model, trainSchema, FeatureColumn.Name); 311public BinaryPredictionTransformer<CalibratedModelParametersBase<FastTreeBinaryModelParameters, PlattCalibrator>> Fit(IDataView trainData, IDataView validationData)
TreeEnsemble\TreeEnsembleCombiner.cs (2)
107var fastTreeModel = new FastTreeBinaryModelParameters(_host, ensemble, featureCount, null); 108return new FeatureWeightsCalibratedModelParameters<FastTreeBinaryModelParameters, PlattCalibrator>(_host, fastTreeModel, cali);
TreeEnsembleFeaturizer.cs (1)
415if (predictor is CalibratedModelParametersBase<FastTreeBinaryModelParameters, PlattCalibrator> calibrated)
Microsoft.ML.IntegrationTests (1)
IntrospectiveTraining.cs (1)
90var fastTreeModel = model.LastTransformer.Model.SubModel;
Microsoft.ML.TestFramework (1)
EnvironmentExtensions.cs (1)
23env.ComponentCatalog.RegisterAssembly(typeof(FastTreeBinaryModelParameters).Assembly); // ML.FastTree
Microsoft.ML.Tests (11)
CalibratedModelParametersTests.cs (9)
82var castedModel = loadedModel as BinaryPredictionTransformer<CalibratedModelParametersBase<FastTreeBinaryModelParameters, PlattCalibrator>>; 86Type expectedInternalType = typeof(FeatureWeightsCalibratedModelParameters<FastTreeBinaryModelParameters, PlattCalibrator>); 95[InlineData(typeof(FastTreeBinaryModelParameters))] 112[InlineData(typeof(FastTreeBinaryModelParameters), false)] 143[InlineData(typeof(FastTreeBinaryModelParameters))] 162[InlineData(typeof(FastTreeBinaryModelParameters))] 203if (modelType == typeof(FastTreeBinaryModelParameters)) 218FastTreeBinaryModelParameters fastTree => new FeatureWeightsCalibratedModelParameters<FastTreeBinaryModelParameters, PlattCalibrator>(Env, fastTree, calibrator),
TrainerEstimators\TreeEnsembleFeaturizerTest.cs (2)
292/// 1. It first trains a <see cref="FastTreeBinaryModelParameters"/> using <see cref="FastTreeBinaryTrainer"/>. 293/// 2. Then, it creates the a <see cref="PretrainedTreeFeaturizationEstimator"/> from the trained <see cref="FastTreeBinaryModelParameters"/>.