1

2024-11-02 21:22:34 +08:00
parent 9d6226e04b
commit 4a8a3dad9b
1 changed files with 0 additions and 12 deletions
@@ -106,15 +106,3 @@ if __name__ == "__main__":
    accuracy_test = accuracy_score(y_test, y_pred_test)
    print(f"Test Accuracy: {accuracy_test}")
    # 随机森林模型
    param_grid = {
        'n_estimators': [100, 200, 300],
        'max_depth': [5, 10, 15],
        'learning_rate': [0.01, 0.05, 0.1]
    }
    rf = RandomForestClassifier()
    grid_search = GridSearchCV(estimator=rf, param_grid=param_grid, cv=5, scoring='accuracy')
    grid_search.fit(X_train, y_train)
    best_rf = grid_search.best_estimator_
    print("Best parameters:", grid_search.best_params_)
    print("Best accuracy:", grid_search.best_score_)