liurui/PriceForecast
Template
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

如果为is_fivemodels为真，不修改最佳模型文档

Browse Source
This commit is contained in:
workpc 2024-12-20 10:49:01 +08:00
parent 2896312a48
commit 53b30ab7a7
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
models/nerulforcastmodels.py
View File

@ -274,6 +274,9 @@ def model_losss(sqlitedb,end_time):
# 保存5个最佳模型的名称 # 保存5个最佳模型的名称
if len(modelnames) > 5: if len(modelnames) > 5:
modelnames = modelnames[0:5] modelnames = modelnames[0:5]
if is_fivemodels:
pass
else:
with open(os.path.join(dataset,"best_modelnames.txt"), 'w') as f: with open(os.path.join(dataset,"best_modelnames.txt"), 'w') as f:
f.write(','.join(modelnames) + '\n') f.write(','.join(modelnames) + '\n')