Boruta_model_pkl_final

# define random forest classifier
forest = RandomForestClassifier(n_jobs=-1, class_weight='balanced', max_depth = 3)
forest.fit(X, Y)

# define Boruta feature selection method
feat_selector = BorutaPy(forest, n_estimators='auto', verbose=0, random_state=1)

model_path = os.path.join("./output/", 'boruta_model_2.pkl')
joblib.dump(feat_selector, model_path)