From ec036307313018e5797fcf6fe22981e56cbda4b2 Mon Sep 17 00:00:00 2001
From: Dengddd <3149942468@qq.com>
Date: Thu, 31 Oct 2024 14:43:36 +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 | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/function.py b/function.py
index cced63e..d81897c 100644
--- a/function.py
+++ b/function.py
@@ -88,17 +88,19 @@ def load_data(filepath):
 
     return data_array[:, :-1], data_array[:, -1]
 
-def preprocess_data(X, y):
+def preprocess_data(data_array):
     """数据预处理，合并特征23的值，并删除特征3和特征24~28。"""
-    # 合并特征23的值：将2（死亡）和3（安乐死）合并为0（死亡）
+    # 取最后一列作为目标变量
+    y = data_array[:, 23]  # 特征23作为目标变量
     y = np.where(y == 2, 0, y)  # 将值2替换为0
     y = np.where(y == 3, 0, y)  # 将值3替换为0
 
     # 删除特征3和特征24~28（假设特征索引从0开始）
-    X = np.delete(X, [2] + list(range(23, 28)), axis=1)
-
+    X = np.delete(data_array, [2] + list(range(24, 29)), axis=1)  # 删除特征3和特征24~28
+    
     return X, y
 
+
 if __name__ == "__main__":
     # 加载数据
     X, y = load_data("./data/horse-colic.data")