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