We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d715b71 commit 7f86ec3Copy full SHA for 7f86ec3
test/Microsoft.ML.Tests/Scenarios/Api/Estimators/IntrospectiveTraining.cs
@@ -65,7 +65,7 @@ public void FastTreeClassificationIntrospectiveTraining()
65
var model = pipeline.Fit(data);
66
67
// Extract the learned GBDT model.
68
- var treeCollection = ((FastForestBinaryModelParameters)((Internal.Calibration.FeatureWeightsCalibratedPredictor)pred.Model).SubPredictor).TrainedTreeEnsemble;
+ var treeCollection = ((FastTreeBinaryModelParameters)((Internal.Calibration.FeatureWeightsCalibratedPredictor)pred.Model).SubPredictor).TrainedTreeEnsemble;
69
70
// Inspect properties in the extracted model.
71
Assert.Equal(3, treeCollection.Trees.Count);
0 commit comments