修改--修改接口文档

This commit is contained in:
haotian 2025-02-19 17:05:10 +08:00
parent 67274d2f1b
commit 04ce95b59f

View File

@ -394,8 +394,9 @@ Response:
"status_url": "/api/train/status/train_20230820_001"
}
```
### 2.5 获取MLFlow中保存的实验
### 2.5 获取已经训练好的模型列表
### 2.6 获取已经训练好的模型列表
```http
GET /api/models/finished/{experiment_name}
@ -404,36 +405,35 @@ Response:
"status": "success",
"models": [
{
"model_id": "model_20250219_001",
"algorithm": "XGBClassifier",
"task_type": "classification",
"dataset": "breast_cancer",
"training_time": "2025-02-19T14:30:00",
"metrics": {
"accuracy": 0.95,
"precision": 0.94,
"recall": 0.93,
"f1": 0.94
},
"parameters": {
"n_estimators": 100,
"learning_rate": 0.1,
"max_depth": 6
},
"mlflow_run_id": "abc123def456",
"model_path": "models/model_20250219_001"
"run_id" 7970364d490f4e0aa0375c2db26215f3
'experiment_id' 656341556838275234
'algorithm': XGBClassifier
'task_type': classification
'dataset': /home/admin-root/haotian/MLPlatform/dataset/dataset_processed/breast_cancer_20250219_144629
'training_start_time': 2025-02-19 08:43:02.067000+00:00
'training_end_time': 2025-02-19 08:43:05.256000+00:00
'metrics':{
roc_auc: 0.9608
recall: 0.9610
f1: 0.9612
precision: 0.9618
accuracy: 0.9610
}
},
{
...
}
],
"total_count": 1,
"total_count": 2,
"page": 1,
"page_size": 10
}
```
### 2.6 删除指定的训练好的模型.
### 2.7 删除指定的训练好的模型.
### 2.7 获取训练状态
### 2.8 获取训练状态
```http
GET /api/train/status/{task_id}