From 9cb82bd06d563d2f75dd5629e430efca3921707d Mon Sep 17 00:00:00 2001
From: tian <11429339@qq.com>
Date: Wed, 10 Jun 2026 12:25:37 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E8=AE=AD=E7=BB=83=E5=90=8E=E6=89=80?=
 =?UTF-8?q?=E6=9C=89=E6=A8=A1=E5=9E=8B=E8=87=AA=E5=8A=A8=E6=BF=80=E6=B4=BB?=
 =?UTF-8?q?;=20=E9=A2=84=E6=B5=8B=E9=A1=B5=E6=97=A0=E9=9C=80=E6=89=8B?=
 =?UTF-8?q?=E5=8A=A8=E6=BF=80=E6=B4=BB=E5=8D=B3=E5=8F=AF=E8=A7=81?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/model_trainer.py | 23 +++++------------------
 1 file changed, 5 insertions(+), 18 deletions(-)

diff --git a/src/model_trainer.py b/src/model_trainer.py
index bc0b833..a38a54c 100644
--- a/src/model_trainer.py
+++ b/src/model_trainer.py
@@ -279,14 +279,7 @@ class ModelTrainer:
             with get_db_connection() as conn:
                 cursor = conn.cursor()
                 
-                # 将所有同类型模型设置为非激活, 除了 PLS 模型
-                cursor.execute("""
-                    UPDATE trained_models 
-                    SET is_active = FALSE 
-                    WHERE equipment_type = ? AND model_type != ?
-                """, (equipment_type, 'pls'))
-                
-                # 保存新模型记录
+                # 保存新模型记录（设置为激活）
                 cursor.execute("""
                     INSERT INTO trained_models (
                         model_name, model_type, equipment_type, model_path, 
@@ -679,19 +672,13 @@ class ModelTrainer:
             with get_db_connection() as conn:
                 cursor = conn.cursor()
                 
-                # 将同类型的其他模型设置为非激活
-                if model_type != 'pls':
+                # PLS：每个装备类型只保留最新一个
+                if model_type == 'pls':
                     cursor.execute("""
                         UPDATE trained_models 
                         SET is_active = FALSE 
-                        WHERE equipment_type = ? AND model_type != ?
-                    """, (equipment_type, 'pls'))
-                else:
-                    cursor.execute("""
-                        UPDATE trained_models 
-                        SET is_active = FALSE 
-                        WHERE equipment_type = ? AND model_type = ?
-                    """, (equipment_type, 'pls'))
+                        WHERE equipment_type = ? AND model_type = 'pls'
+                    """, (equipment_type,))
                 
                 # 保存新模型记录
                 cursor.execute("""