更新 function.py

function.py

@@ -109,7 +109,6 @@ def cross_validate(X, y, k=5, learning_rate=0.01, iterations=1000):
     return np.mean(accuracies)
 
 if __name__ == "__main__":
-    # 以下代码可用于测试，可自行随意修改
     # 加载训练集
     X_train, y_train = load_data("./data/horse-colic.data")
 
@@ -117,7 +116,7 @@ if __name__ == "__main__":
     X_test, y_test = load_data("./data/horse-colic.test")
 
     # 训练对率回归模型
-     k = 5  # 设置折数
-    accuracy = cross_validate(X, y, k=k, learning_rate=0.01, iterations=5000)
+    k = 5  # 设置折数
+    accuracy = cross_validate(X_train, y_train, k=k, learning_rate=0.01, iterations=5000)
 
     print(f"{k}-折交叉验证的平均准确率: {accuracy:.2f}")