From 15f20bc9910a20b02ad1cb250e5be07bb11a3384 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?xavier=20dupr=C3=A9?= Date: Sat, 25 Apr 2020 13:01:36 +0200 Subject: [PATCH] Update ml_featurizer.py --- mlinsights/mlmodel/ml_featurizer.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/mlinsights/mlmodel/ml_featurizer.py b/mlinsights/mlmodel/ml_featurizer.py index 58e2912e..d1d4ffd4 100644 --- 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):