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()
-