添加--接口文档中添加获取训练好的模型方法
This commit is contained in:
parent
f19a56f90b
commit
d214d881c2
@ -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
|
||||
|
||||
Loading…
Reference in New Issue
Block a user