修改--修改根据实验名返回模型管理方法的返回参数
This commit is contained in:
parent
c0a30795c9
commit
67274d2f1b
@ -397,7 +397,7 @@ Response:
|
||||
|
||||
### 2.5 获取已经训练好的模型列表
|
||||
```http
|
||||
GET /api/models/finished
|
||||
GET /api/models/finished/{experiment_name}
|
||||
|
||||
Response:
|
||||
{
|
||||
|
||||
@ -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']}")
|
||||
|
||||
Binary file not shown.
@ -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'], # 从配置或其他地方获取
|
||||
|
||||
Loading…
Reference in New Issue
Block a user