更新 function.py

function.py

@@ -62,12 +62,17 @@ def load_data(filepath):
     return X, y
 
 if __name__ == "__main__":
-    # 以下代码可用于测试，可自行随意修改
     # 加载训练集
     X_train, y_train = load_data("./data/horse-colic.data")
+    if X_train is None or y_train is None:
+        print("训练数据加载失败，程序终止。")
+        exit(1)
 
     # 加载测试集
     X_test, y_test = load_data("./data/horse-colic.test")
+    if X_test is None or y_test is None:
+        print("测试数据加载失败，程序终止。")
+        exit(1)
 
     # 训练支持向量机模型
     weights, bias = train_svm(X_train, y_train, learning_rate=0.01, iterations=1000)