<s> import data_io <EOL> from sklearn . ensemble import RandomForestClassifier <EOL> def main ( ) : <EOL> print ( "<STR_LIT>" ) <EOL> train = data_io . read_train ( ) <EOL> train . fillna ( <NUM_LIT:0> , inplace = True ) <EOL> train_sample = train [ : <NUM_LIT> ] . fillna ( value = <NUM_LIT:0> ) <EOL> feature_names = list ( train_sample . columns ) <EOL> feature_names . remove ( "<STR_LIT>" ) <EOL> feature_names . remove ( "<STR_LIT>" ) <EOL> feature_names . remove ( "<STR_LIT>" ) <EOL> feature_names . remove ( "<STR_LIT>" ) <EOL> feature_names . remove ( "<STR_LIT>" ) <EOL> features = train_sample [ feature_names ] . values <EOL> target = train_sample [ "<STR_LIT>" ] . values <EOL> print ( "<STR_LIT>" ) <EOL> classifier = RandomForestClassifier ( n_estimators = <NUM_LIT:50> , <EOL> verbose = <NUM_LIT:2> , <EOL> n_jobs = <NUM_LIT:1> , <EOL> min_samples_split = <NUM_LIT:10> , <EOL> random_state = <NUM_LIT:1> ) <EOL> classifier . fit ( features , target ) <EOL> print ( "<STR_LIT>" ) <EOL> data_io . save_model ( classifier ) <EOL> if __name__ == "<STR_LIT:__main__>" : <EOL> main ( ) </s>
