51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
"""
|
|||
|
|
最小训练测试 - 直接调用训练器
|
|||
|
|
"""
|
|||
|
|
import sys
|
|||
|
|
import os
|
|||
|
|
sys.path.append(os.path.join(os.path.dirname(__file__), 'server'))
|
|||
|
|
|
|||
|
|
def minimal_training_test():
|
|||
|
|
"""直接调用训练器进行测试"""
|
|||
|
|
print("=== 最小训练测试 ===")
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 导入必要的模块
|
|||
|
|
from trainers.mlstm_trainer import train_product_model_with_mlstm
|
|||
|
|
|
|||
|
|
print("✅ 训练器导入成功")
|
|||
|
|
|
|||
|
|
# 创建一个模拟的socketio对象
|
|||
|
|
class MockSocketIO:
|
|||
|
|
def emit(self, event, data, namespace=None):
|
|||
|
|
print(f"[MockSocketIO] {event}: {data}")
|
|||
|
|
|
|||
|
|
mock_socketio = MockSocketIO()
|
|||
|
|
|
|||
|
|
print("🚀 开始最小训练测试（2个epoch）...")
|
|||
|
|
|
|||
|
|
# 直接调用训练器
|
|||
|
|
result = train_product_model_with_mlstm(
|
|||
|
|
product_id="P001",
|
|||
|
|
epochs=2, # 只训练2个epoch
|
|||
|
|
socketio=mock_socketio,
|
|||
|
|
task_id="test-direct-123"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
print(f"✅ 训练完成: {type(result)}")
|
|||
|
|
if isinstance(result, tuple) and len(result) >= 2:
|
|||
|
|
model, metrics = result[:2]
|
|||
|
|
print(f"✅ 返回结果: 模型={type(model)}, 指标={metrics}")
|
|||
|
|
else:
|
|||
|
|
print(f"⚠️ 返回结果格式异常: {result}")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"❌ 最小训练测试失败: {e}")
|
|||
|
|
import traceback
|
|||
|
|
traceback.print_exc()
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
print("🧪 开始最小训练测试")
|
|||
|
|
minimal_training_test()
|
|||
|
|
print("🏁 测试完成")
|