From 5169ad63aa540ebadec0ae199319386cb6d8714d Mon Sep 17 00:00:00 2001
From: jack <2170067548@qq.com>
Date: Sun, 3 Nov 2024 17:20:12 +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 | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/function.py b/function.py
index b6bd3c4..21c3d82 100644
--- a/function.py
+++ b/function.py
@@ -40,7 +40,9 @@ def load_and_preprocess_horse_colic_data(train_filename, test_filename):
     # 定义列名
     column_names = [
         'surgery', 'age', 'hospital_number', 'rectal_temperature',
-        # ... 其他列名可以根据需要添加
+        'pulse', 'respiratory_rate', 'temperature_of_extremities', 
+        'peripheral_pulse', 'mucous_membrane', 'capillary_refill_time',
+        'pain', 'surgical_lesion', 'type_of_lesion', 'surgical_lesion_location',
         'outcome'  # 假设'outcome'是目标变量，1=lived, 2=died, 3=euthanized
     ]
 
@@ -90,4 +92,3 @@ def main():
 
 if __name__ == "__main__":
     main()
-