From da55d99f08d92c52dc06325fcef9920aeb72e0d1 Mon Sep 17 00:00:00 2001
From: zhangchenghong <1598349064@qq.com>
Date: Thu, 14 Nov 2024 17:21:16 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20function.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 function.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/function.py b/function.py
index c2f90ea..a670cbc 100644
--- a/function.py
+++ b/function.py
@@ -56,10 +56,9 @@ def load_data(filepath):
     X = data[:, :-1]  # 特征数据（假设最后一列是目标变量）
     y = data[:, 23]   # 特征23作为目标变量
     
-    # 转换目标变量为 1 (lived), 2 (died), 3 (euthanized)
-    y = np.where(y == 2, 0, y)  # 2 -> 0 (died -> death)
-    y = np.where(y == 3, 0, y)  # 3 -> 0 (euthanized -> death)
-
+    # 不合并死亡与安乐死，直接使用原始标签
+    # 1 = lived, 2 = died, 3 = euthanized
+    
     # 将缺失值填充为 0（或其他合适的值）
     X = np.nan_to_num(X, nan=0)
     
@@ -122,4 +121,4 @@ if __name__ == "__main__":
             correct += 1
     
     accuracy = correct / len(X_test) * 100
-    print(f"测试集准确率: {accuracy}%")
+    print(f"测试集准确率: {accuracy}%")
\ No newline at end of file