修改--修改根据实验名返回模型管理方法的返回参数

This commit is contained in:
haotian 2025-02-19 17:00:16 +08:00
parent c0a30795c9
commit 67274d2f1b
4 changed files with 9 additions and 3 deletions

View File

@ -397,7 +397,7 @@ Response:
### 2.5 获取已经训练好的模型列表
```http
GET /api/models/finished
GET /api/models/finished/{experiment_name}
Response:
{

View File

@ -19,7 +19,12 @@ if result['status'] == 'success':
print(f"每页数量: {result['page_size']}")
print("\n模型列表:")
for model in result['models']:
print(f"\n模型ID: {model['model_id']}")
'''
'run_id': run['run_id'],
'experiment_id': run['experiment_id'],
'''
print(f"run_id", model['run_id'])
print(f"experiment_id", model['experiment_id'])
print(f"算法: {model['algorithm']}")
print(f"任务类型: {model['task_type']}")
print(f"数据集: {model['dataset']}")

View File

@ -96,7 +96,8 @@ class ModelManager:
# 构建模型信息
model_info = {
'model_id': f"model_{pd.to_datetime(run['start_time']).strftime('%Y%m%d_%H%M%S')}",
'run_id': run['run_id'],
'experiment_id': run['experiment_id'],
'algorithm': params['algorithm'], # 从配置或其他地方获取
'task_type': params['task_type'], # 从配置或其他地方获取
'dataset': params['dataset'], # 从配置或其他地方获取