diff --git a/sklearn_porter/estimator/classifier/RandomForestClassifier/__init__.py b/sklearn_porter/estimator/classifier/RandomForestClassifier/__init__.py index 83bf3212..314c374a 100644 --- a/sklearn_porter/estimator/classifier/RandomForestClassifier/__init__.py +++ b/sklearn_porter/estimator/classifier/RandomForestClassifier/__init__.py @@ -275,7 +275,7 @@ def create_single_method(self, estimator_index, estimator): :return : string The created method. """ - indices = [self.repr(e) for e in estimator.tree_.feature] + indices = [str(e) for e in estimator.tree_.feature] tree_branches = self.create_branches( estimator.tree_.children_left, estimator.tree_.children_right,