63 lines
2.1 KiB
Python
63 lines
2.1 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
测试训练日志输出
|
|||
|
|
"""
|
|||
|
|
import requests
|
|||
|
|
import json
|
|||
|
|
import time
|
|||
|
|
|
|||
|
|
def test_training_logs():
|
|||
|
|
"""测试训练API和日志输出"""
|
|||
|
|
|
|||
|
|
# 1. 启动训练任务
|
|||
|
|
print("=== 启动训练任务 ===")
|
|||
|
|
training_data = {
|
|||
|
|
"product_id": "P001",
|
|||
|
|
"model_type": "mlstm",
|
|||
|
|
"epochs": 3, # 只训练3个epoch进行测试
|
|||
|
|
"training_mode": "product"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
response = requests.post(
|
|||
|
|
'http://localhost:5000/api/training/start',
|
|||
|
|
json=training_data,
|
|||
|
|
headers={'Content-Type': 'application/json'}
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
if response.status_code == 200:
|
|||
|
|
result = response.json()
|
|||
|
|
task_id = result.get('task_id')
|
|||
|
|
print(f"✅ 训练任务已启动，任务ID: {task_id}")
|
|||
|
|
|
|||
|
|
# 2. 监控训练状态
|
|||
|
|
print("\n=== 监控训练状态 ===")
|
|||
|
|
for i in range(30): # 最多监控30次
|
|||
|
|
time.sleep(2)
|
|||
|
|
|
|||
|
|
status_response = requests.get(f'http://localhost:5000/api/training/status/{task_id}')
|
|||
|
|
if status_response.status_code == 200:
|
|||
|
|
status_data = status_response.json()
|
|||
|
|
status = status_data.get('status', 'unknown')
|
|||
|
|
print(f"第{i+1}次检查 - 状态: {status}")
|
|||
|
|
|
|||
|
|
if status == 'completed':
|
|||
|
|
print("✅ 训练已完成!")
|
|||
|
|
print(f"指标: {status_data.get('metrics', 'None')}")
|
|||
|
|
break
|
|||
|
|
elif status == 'failed':
|
|||
|
|
print("❌ 训练失败!")
|
|||
|
|
print(f"错误: {status_data.get('error', 'Unknown')}")
|
|||
|
|
break
|
|||
|
|
else:
|
|||
|
|
print(f"⚠️ 无法获取状态: {status_response.status_code}")
|
|||
|
|
|
|||
|
|
else:
|
|||
|
|
print(f"❌ 启动训练失败: {response.status_code}")
|
|||
|
|
print(f"响应: {response.text}")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"❌ 测试失败: {e}")
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
test_training_logs()
|