2 writes to _ensemble
Microsoft.ML.FastTree (2)
TreeEnsembleFeaturizer.cs (2)
417
_ensemble
= predictor as TreeEnsembleModelParameters;
436
ctx.LoadModel<TreeEnsembleModelParameters, SignatureLoadModel>(env, out
_ensemble
, "Ensemble");
13 references to _ensemble
Microsoft.ML.FastTree (13)
TreeEnsembleFeaturizer.cs (13)
118
var treeValueType = new VectorDataViewType(NumberDataViewType.Single, owner.
_ensemble
.TrainedEnsemble.NumTrees);
129
var pathIdType = new VectorDataViewType(NumberDataViewType.Single, owner._totalLeafCount - owner.
_ensemble
.TrainedEnsemble.NumTrees);
189
var state = new State(_ectx, input, _owner.
_ensemble
, _owner._totalLeafCount, FeatureColumn.Index);
418
_host.Check(
_ensemble
!= null, "Predictor in model file does not have compatible type");
420
_totalLeafCount = CountLeaves(
_ensemble
);
437
_totalLeafCount = CountLeaves(
_ensemble
);
455
_host.AssertValue(
_ensemble
);
456
ctx.SaveModel(
_ensemble
, "Ensemble");
476
var numTrees =
_ensemble
.TrainedEnsemble.NumTrees;
487
var numTrees =
_ensemble
.TrainedEnsemble.NumTrees;
492
foreach (var tree in ((ITreeEnsemble)
_ensemble
).GetTrees())
504
var numTrees =
_ensemble
.TrainedEnsemble.NumTrees;
511
foreach (var tree in ((ITreeEnsemble)
_ensemble
).GetTrees())