更新 function.py

2024-11-05 11:08:15 +08:00
parent 173c40cedd
commit f699d7ea09
1 changed files with 1 additions and 2 deletions
@@ -86,9 +86,8 @@ def load_data(filepath):
    X = data[:, :-1]  # 假设最后一列为标签  
    y = data[:, -1]   # 最后一列为标签  

-    # 确保 y 是二进制的，1表示死亡，0表示存活  
    y = np.where(y == 1, 1, 0)  # 处理标签，确保其为0或1  
-    return X, y  
+    return X, y   


 if __name__ == "__main__":