diff --git a/function.py b/function.py
index 8cb28e1..f7ad8e3 100644
--- a/function.py
+++ b/function.py
@@ -90,6 +90,11 @@ def load_data(filepath):
     - np.array: 标签数据。
     """
     data = np.loadtxt(filepath, delimiter=',', dtype=str)
+    
+    # 检查数据的维度
+    if data.ndim == 1:
+        data = data.reshape(1, -1)  # 如果数据是一维的，重塑为二维
+
     X = data[:, :-1].astype(float)  # 特征数据
     y = data[:, -1]  # 标签数据