Fix for C RandomForestClassifier code generation.

Don't apply num_format conversion in RandomForestClassifier code generation for feature indices.
jeroenrook · Dec 11, 2018 · 43a403c · 43a403c
1 parent f021c46
commit 43a403c
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
@@ -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,