Teste de nova restrição

SRLib/Utilities.py

@@ -1,5 +1,4 @@
 ### Imports
-from operator import index
 import pickle as pkl
 
 
@@ -19,11 +18,11 @@ class encoding():
         mms_pclass = pkl.load( open( './models/mms_pclass.sav', 'rb' ) )
 
         df = data_raw[[ 'Sex']]        
-        df['__Mr']    =  1 if( (data_raw['Sex'] == 0 ) & (data_raw['Age'] > 20)) else 0
         df['Age']     = mms_age.transform( data_raw[['Age']].values )
         df['SibSp']   = mms_sibsp.transform( data_raw[['SibSp']].values )
         df['Fare']    = mms_fare.transform( data_raw[['Fare']].values )
         df['Pclass']  = mms_pclass.transform( data_raw[['Pclass']].values ) 
+        df['__Mr']    = df.Sex.apply( lambda x : 1 if x == 0 else 0) 
 
         ### Feature Selection
         df = df[['Fare', 'Age', 'Sex', 'Pclass', '__Mr', 'SibSp']]