更新 function.py

2024-11-21 17:21:52 +08:00
parent eb6c2cb8e2
commit 6aca57876b
1 changed files with 1 additions and 1 deletions
@@ -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