diff --git a/sklearn_porter/estimator/classifier/RandomForestClassifier/__init__.py b/sklearn_porter/estimator/classifier/RandomForestClassifier/__init__.py index 3fab61d9..9e449a82 100644 --- a/sklearn_porter/estimator/classifier/RandomForestClassifier/__init__.py +++ b/sklearn_porter/estimator/classifier/RandomForestClassifier/__init__.py @@ -277,7 +277,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,