Update ml_featurizer.py

sdpython · Apr 25, 2020 · 15f20bc · 15f20bc
1 parent f2b06ae
commit 15f20bc
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/mlinsights/mlmodel/ml_featurizer.py b/mlinsights/mlmodel/ml_featurizer.py
@@ -120,17 +120,17 @@ def model_featurizer_rfc(model, output=True):
     @return                 function
     """
     if output:
-        def feat(X, model, many):
+        def feat1(X, model, many):
             "wraps sklearn"
             return wrap_predict_sklearn(X, model.predict_proba, many)
 
-        return lambda X, many, model=model: feat(X, model, many)
-
-    def feat(X, model, many):
+        return lambda X, many, model=model: feat1(X, model, many)
+    
+    def feat2(X, model, many):
         "wraps sklearn"
         return wrap_predict_sklearn(X, model.apply, many)
 
-    return lambda X, many, model=model: feat(X, model, many)
+    return lambda X, many, model=model: feat2(X, model, many)
 
 
 def wrap_predict_keras(X, fct, many, shapes):