更新 function.py

function.py

@@ -86,10 +86,14 @@ def load_data(filepath):
     data = np.genfromtxt(filepath, delimiter=',', dtype=float, filling_values=np.nan)
     # 移除含有NaN的行
     data = data[~np.isnan(data).any(axis=1)]
+    # 确保数据至少有两列，否则无法分割为特征和标签
+    if data.shape[1] < 2:
+        raise ValueError("数据文件格式不正确，应至少包含两列")
     X = data[:, :-1]
     y = data[:, -1].astype(int)
     return X, y
 
+
 if __name__ == "__main__":
     # 以下代码可用于测试，可自行随意修改
     # 加载训练集