Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- Column A Column B AnnualFee ColumnD ColumnE
- 4500 3.9 5% 2.1 7
- 4500 3.9 10% 2.1 7
- 4500 3.9 15% 2.1 7
- 4500 3.9 20% 2.1 7
- 4500 3.9 25% 2.1 7
- rf = RandomForestClassifier(n_estimators = 400,random_state = 0,
- min_samples_split=2,min_samples_leaf=5,
- class_weight = {0:.0001,1:.9999})
- rf.fit(X_train, Y_train )
- from sklearn.model_selection import StratifiedKFold
- from sklearn.model_selection import GridSearchCV
- clf_4 = RandomForestClassifier(class_weight = {0:1,1:5})
- estimators_range = np.array([2,3,4,5,6,7,8,9,10,15,20,25])
- depth_range = np.array([11,21,35,51,75,101,151,201,251,301,401,451,501])
- kfold = 5
- skf = StratifiedKFold(n_splits = kfold,random_state = 42)
- model_grid = [{'max_depth': depth_range, 'n_estimators': estimators_range}]
- grid = GridSearchCV(clf_4, model_grid, cv = StratifiedKFold(n_splits = 5,
- random_state = 42),n_jobs = 8, scoring = 'roc_auc')
- grid.fit(X_train,Y_train)
Add Comment
Please, Sign In to add comment