更新 function.py

function.py

@@ -135,9 +135,14 @@ def grid_search(X_train, y_train, X_test, y_test, param_grid):
     for params in product(*param_grid.values()):
         learning_rate, iterations, batch_size, regularization, lambda_reg = params
 
+        # 训练模型
         weights = train_logreg(X_train, y_train, learning_rate, iterations, batch_size, regularization, lambda_reg)
+
+        # 计算测试集准确率
         acc = accuracy(X_test, y_test, weights)
 
+        print(f"尝试参数组合: {params}, 测试集准确率: {acc:.4f}")
+
         if acc > best_accuracy:
             best_accuracy = acc
             best_params = params
@@ -145,8 +150,8 @@ def grid_search(X_train, y_train, X_test, y_test, param_grid):
     return best_params, best_accuracy
 
 if __name__ == "__main__":
-    X_train, y_train = load_data("./horse-colic.data")
-    X_test, y_test = load_data("./horse-colic.test")
+    X_train, y_train = load_data("./data/horse-colic.data")
+    X_test, y_test = load_data("./data/horse-colic.test")
 
     # 定义参数网格
     param_grid = {