Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- from sklearn.pipeline import Pipeline
- from sklearn.preprocessing import StandardScaler
- from sklearn.decomposition import PCA
- from sklearn.ensemble import RandomForestClassifier
- pipeline = Pipeline([
- ('scaler', StandardScaler()),
- ('pca', PCA()),
- ('classifier', RandomForestClassifier())
- ])
- from sklearn.model_selection import GridSearchCV
- param_grid = {
- 'pca__n_components': [2, 5, 10],
- 'classifier__n_estimators': [50, 100, 200],
- 'classifier__max_depth': [None, 10, 20, 30]
- }
- grid_search = GridSearchCV(pipeline, param_grid, cv=5)
- grid_search.fit(X_train, y_train)
- print("Best parameters:", grid_search.best_params_)
- print("Best score:", grid_search.best_score_)
Advertisement
Add Comment
Please, Sign In to add comment