Fix IsolationForest with max_samples="auto" (#327)

dmlc · Jan 4, 2022 · 0877988 · 0877988
1 parent 5d80909
commit 0877988
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/treelite/sklearn/importer.py b/python/treelite/sklearn/importer.py
@@ -132,7 +132,7 @@ def import_model(sklearn_model):
         raise TreeliteError("Gradient boosted trees must be trained with the option init='zero'")
 
     if isinstance(sklearn_model, IsolationForest):
-        ratio_c = expected_depth(sklearn_model.max_samples)
+        ratio_c = expected_depth(sklearn_model.max_samples_)
 
     node_count = []
     children_left = ArrayOfArrays(dtype=np.int64)