From 6a785ba0586428f0dbc25004a6efc6076d1c429e Mon Sep 17 00:00:00 2001
From: zhangchenghong <1598349064@qq.com>
Date: Fri, 15 Nov 2024 14:51: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 | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/function.py b/function.py
index a670cbc..d25ea83 100644
--- a/function.py
+++ b/function.py
@@ -54,7 +54,7 @@ def load_data(filepath):
 
     # 选择特征和标签
     X = data[:, :-1]  # 特征数据（假设最后一列是目标变量）
-    y = data[:, 23]   # 特征23作为目标变量
+    y = data[:, -1]   # 最后一列作为目标变量
     
     # 不合并死亡与安乐死，直接使用原始标签
     # 1 = lived, 2 = died, 3 = euthanized
@@ -81,7 +81,7 @@ def train_svm(X, y):
     X_scaled = scaler.fit_transform(X)
     
     # 使用RBF核的SVM模型
-    model = SVC(kernel='rbf', decision_function_shape='ovr')  # 采用RBF核
+    model = SVC(kernel='rbf', decision_function_shape='ovo')  # 采用RBF核，多类分类使用'ovo'
     model.fit(X_scaled, y)
     
     return model, scaler