4 instantiations of FastTreeBinaryModelParameters
Microsoft.ML.FastTree (4)
FastTreeClassification.cs (2)
92var predictor = new FastTreeBinaryModelParameters(env, ctx); 206var 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);
23 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); 97return new SchemaBindableCalibratedModelParameters<FastTreeBinaryModelParameters, ICalibrator>(env, predictor, calibrator); 131BinaryPredictionTransformer<CalibratedModelParametersBase<FastTreeBinaryModelParameters, PlattCalibrator>>, 132CalibratedModelParametersBase<FastTreeBinaryModelParameters, PlattCalibrator>> 184private protected override CalibratedModelParametersBase<FastTreeBinaryModelParameters, PlattCalibrator> TrainModelCore(TrainContext context) 206var pred = new FastTreeBinaryModelParameters(Host, TrainedEnsemble, FeatureCount, InnerOptions); 213return new FeatureWeightsCalibratedModelParameters<FastTreeBinaryModelParameters, PlattCalibrator>(Host, pred, cali); 301private protected override BinaryPredictionTransformer<CalibratedModelParametersBase<FastTreeBinaryModelParameters, PlattCalibrator>> MakeTransformer( 302CalibratedModelParametersBase<FastTreeBinaryModelParameters, PlattCalibrator> model, DataViewSchema trainSchema) 303=> new BinaryPredictionTransformer<CalibratedModelParametersBase<FastTreeBinaryModelParameters, PlattCalibrator>>(Host, model, trainSchema, FeatureColumn.Name); 309public 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 (4)
CalibratedModelParametersTests.cs (2)
79var castedModel = loadedModel as BinaryPredictionTransformer<CalibratedModelParametersBase<FastTreeBinaryModelParameters, PlattCalibrator>>; 83Type expectedInternalType = typeof(FeatureWeightsCalibratedModelParameters<FastTreeBinaryModelParameters, PlattCalibrator>);
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"/>.