更新 function.py

function.py

@@ -145,11 +145,8 @@ def k_fold_cross_validation(X, y, k=5, learning_rate=0.001, iterations=10000, ba
 
 if __name__ == "__main__":
     # 加载训练集
-    try:
+    X_train, y_train = load_data("./data/horse-colic.data")
-        X_train, y_train = load_data("./data/horse-colic.data")
+
-    except FileNotFoundError as e:
-        print(e)
-        exit()
 
     # 数据归一化处理
     scaler = StandardScaler()
@@ -160,11 +157,8 @@ if __name__ == "__main__":
     print(f"K折交叉验证的平均准确率: {avg_accuracy:.4f}")
 
     # 加载测试集
-    try:
+    X_test, y_test = load_data("./data/horse-colic.test")
-        X_test, y_test = load_data("./data/horse-colic.test")
+
-    except FileNotFoundError as e:
-        print(e)
-        exit()
 
     # 测试集也需要进行相同的归一化
     X_test = scaler.transform(X_test)  # 使用相同的Scaler进行测试集的标准化