tian/CostPrediction
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: 训练后所有模型自动激活; 预测页无需手动激活即可见

Browse Source
This commit is contained in:
tian 2026-06-10 12:25:37 +08:00
parent 42891d2980
commit 9cb82bd06d
1 changed files with 5 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
src/model_trainer.py
View File

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