ajoubrel-ensae/BDC-team-1
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix time to open

Browse Source
This commit is contained in:
Alexis REVELLE 2024-03-28 13:17:45 +00:00
parent d5ab3c2d68
commit 7ed8516009
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
utils_ml.py
View File

@ -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':
numeric_features.pop('time_to_open')
"""
numeric_features.remove('time_to_open')
if type_of_model=='premium':
if type_of_activity=='musique':
binary_features.extend(['target_optin', 'target_newsletter'])