1

function.py

@@ -42,7 +42,6 @@ def load_data(filepath):
         del data[26]
         del data[25]
         del data[24]
-        del data[23]
         del data[22]
         del data[2]
         for i in data_25:
@@ -107,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_)