添加--接口文档中添加获取训练好的模型方法

This commit is contained in:
haotian 2025-02-19 15:36:09 +08:00
parent f19a56f90b
commit d214d881c2

View File

@ -395,7 +395,45 @@ Response:
}
```
### 2.5 获取训练状态
### 2.5 获取已经训练好的模型列表
```http
GET /api/models/finished
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"
}
],
"total_count": 1,
"page": 1,
"page_size": 10
}
```
### 2.6 删除指定的训练好的模型.
### 2.7 获取训练状态
```http
GET /api/train/status/{task_id}
@ -418,7 +456,7 @@ Response:
}
```
### 2.6 模型预测
### 2.8 模型预测
```http
POST /api/predict
Content-Type: application/json