Advertisement
Guest User

Untitled

a guest
Oct 28th, 2016
67
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.52 KB | None | 0 0
  1. kf = StratifiedKFold(n_splits = 5, random_state=42)
  2. param = {"criterion":["gini", "entropy"], "max_features":[1, 2, 3, 4, 5, 10, 15, 20, 25, 30, 35, 40, "log2", "auto"],
  3. "max_depth":[2, 4, 8, 16, 32, 64], "class_weight":[None, "balanced"]}
  4. cv = GridSearchCV(RandomForestClassifier(), param, scoring="accuracy", n_jobs = -1, cv=kf)
  5. cv.fit(X_train, y_train);
  6. print("Best score (accuracy): ", cv.best_score_)
  7. print("Best params: ", cv.best_params_)
  8. print("Score on X_test: ", accuracy_score(y_test, cv.predict(X_test)))
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement