diff --git a/function.py b/function.py
index ca328f9..048c2d7 100644
--- a/function.py
+++ b/function.py
@@ -108,7 +108,7 @@ def load_data(filepath):
     y = np.where(y == '0', 0, y)  # 假设 '0' 表示存活
 
     # 确保 y 是整数类型
-    y = y.astype(int)
+    y = y.astype(int)  # 将 y 转换为整数类型
 
     return X, y