From 840ce876e2a38438eb0db57bfb2fe2081769a7c1 Mon Sep 17 00:00:00 2001 From: arevelle-ensae Date: Thu, 28 Mar 2024 11:37:33 +0000 Subject: [PATCH] fix preproc --- utils_ml.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/utils_ml.py b/utils_ml.py index 825f9c0..850a162 100644 --- a/utils_ml.py +++ b/utils_ml.py @@ -114,10 +114,10 @@ def preprocess(type_of_model, type_of_activity): binary_features = ['gender_female', 'gender_male', 'achat_internet', 'categorie_age_0_10', 'categorie_age_10_20', 'categorie_age_20_30','categorie_age_30_40', 'categorie_age_40_50', 'categorie_age_50_60', 'categorie_age_60_70', 'categorie_age_70_80', 'categorie_age_plus_80','categorie_age_inconnue', 'country_fr', 'is_profession_known', 'is_zipcode_known', 'opt_in'] - + """ if type_of_activity=='musee': - binary_features.pop('time_to_open') - + numeric_features.pop('time_to_open') + """ if type_of_model=='premium': if type_of_activity=='musique': binary_features.extend(['target_optin', 'target_newsletter'])