release/0.7.0: Merge branch 'master' into release/0.7.0

nok · Dec 15, 2018 · fb16028 · fb16028
2 parents b42f3cc + b378db7
commit fb16028
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,