Skip to content

Commit 7f86ec3

Browse files
committed
Fix typo
1 parent d715b71 commit 7f86ec3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/Microsoft.ML.Tests/Scenarios/Api/Estimators/IntrospectiveTraining.cs

+1-1
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ public void FastTreeClassificationIntrospectiveTraining()
6565
var model = pipeline.Fit(data);
6666

6767
// Extract the learned GBDT model.
68-
var treeCollection = ((FastForestBinaryModelParameters)((Internal.Calibration.FeatureWeightsCalibratedPredictor)pred.Model).SubPredictor).TrainedTreeEnsemble;
68+
var treeCollection = ((FastTreeBinaryModelParameters)((Internal.Calibration.FeatureWeightsCalibratedPredictor)pred.Model).SubPredictor).TrainedTreeEnsemble;
6969

7070
// Inspect properties in the extracted model.
7171
Assert.Equal(3, treeCollection.Trees.Count);

0 commit comments

Comments
 (0)