115 lines
3.7 KiB
Python
115 lines
3.7 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
直接控制台测试 - 验证编码和表情符号显示
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
# 设置环境变量
|
|||
|
|
os.environ['PYTHONIOENCODING'] = 'utf-8'
|
|||
|
|
|
|||
|
|
# 添加server路径
|
|||
|
|
sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'server'))
|
|||
|
|
|
|||
|
|
def test_console_output():
|
|||
|
|
"""测试控制台输出"""
|
|||
|
|
|
|||
|
|
print("=" * 60)
|
|||
|
|
print("🧪 控制台编码测试开始")
|
|||
|
|
print("=" * 60)
|
|||
|
|
|
|||
|
|
# 测试基本中文和表情符号
|
|||
|
|
print("📝 基本测试:")
|
|||
|
|
print(" ✅ 简体中文: 药店销售预测系统")
|
|||
|
|
print(" 🚀 表情符号: 启动 📊 数据 🤖 模型")
|
|||
|
|
print(" 💾 混合文本: Product P001 - 感冒灵颗粒")
|
|||
|
|
|
|||
|
|
# 测试训练相关输出
|
|||
|
|
print("\n🏃 训练日志模拟:")
|
|||
|
|
task_id = "test-12345"
|
|||
|
|
print(f"🚀 训练任务开始: {task_id}")
|
|||
|
|
print(f"📋 任务详情: 训练 TRANSFORMER 模型 - 药品 P001")
|
|||
|
|
print(f"⚙️ 配置参数: 共 3 个轮次")
|
|||
|
|
print(f"🤖 调用 TRANSFORMER 训练器 - 产品: P001")
|
|||
|
|
|
|||
|
|
# 模拟训练进度
|
|||
|
|
for epoch in range(1, 4):
|
|||
|
|
loss = 0.1234 - epoch * 0.01
|
|||
|
|
print(f"📊 Epoch {epoch}/3, 训练损失: {loss:.4f}, 测试损失: {loss+0.01:.4f}")
|
|||
|
|
|
|||
|
|
print(f"✅ 训练完成!")
|
|||
|
|
print(f"📈 训练指标: {{'mse': 0.0123, 'rmse': 0.1107, 'r2': 0.95}}")
|
|||
|
|
print(f"💾 模型保存路径: saved_models/transformer/P001_v1.0.pth")
|
|||
|
|
print(f"✔️ 任务状态更新: 已完成")
|
|||
|
|
|
|||
|
|
print("\n" + "=" * 60)
|
|||
|
|
print("🎉 控制台编码测试完成")
|
|||
|
|
print("💡 如果上述中文和表情符号都能正确显示，说明编码配置成功")
|
|||
|
|
print("=" * 60)
|
|||
|
|
|
|||
|
|
def test_direct_trainer():
|
|||
|
|
"""直接测试训练器"""
|
|||
|
|
|
|||
|
|
print("\n" + "=" * 60)
|
|||
|
|
print("🔧 直接训练器测试")
|
|||
|
|
print("=" * 60)
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 导入预测器
|
|||
|
|
print("📦 导入预测器模块...")
|
|||
|
|
from core.predictor import PharmacyPredictor
|
|||
|
|
|
|||
|
|
print("✅ 预测器导入成功")
|
|||
|
|
|
|||
|
|
# 创建实例
|
|||
|
|
predictor = PharmacyPredictor()
|
|||
|
|
|
|||
|
|
if predictor.data is not None:
|
|||
|
|
print(f"📊 数据加载成功，行数: {len(predictor.data)}")
|
|||
|
|
|
|||
|
|
# 获取产品列表
|
|||
|
|
products = predictor.data['product_id'].unique()
|
|||
|
|
print(f"🎯 可用产品: {list(products[:3])}...") # 只显示前3个
|
|||
|
|
|
|||
|
|
# 选择产品进行快速测试
|
|||
|
|
test_product = products[0]
|
|||
|
|
print(f"\n🚀 开始训练测试 - 产品: {test_product}")
|
|||
|
|
print("=" * 40)
|
|||
|
|
|
|||
|
|
# 调用训练（少量轮次）
|
|||
|
|
metrics = predictor.train_model(
|
|||
|
|
product_id=test_product,
|
|||
|
|
model_type='transformer',
|
|||
|
|
epochs=1, # 只训练1轮快速测试
|
|||
|
|
training_mode='product'
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
print("=" * 40)
|
|||
|
|
if metrics:
|
|||
|
|
print("✅ 训练成功完成")
|
|||
|
|
print(f"📊 返回指标: {metrics}")
|
|||
|
|
else:
|
|||
|
|
print("⚠️ 训练返回None，可能是数据不足")
|
|||
|
|
|
|||
|
|
else:
|
|||
|
|
print("❌ 数据加载失败")
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"❌ 训练测试失败: {e}")
|
|||
|
|
# 不打印完整traceback，只显示错误信息
|
|||
|
|
|
|||
|
|
print("\n" + "=" * 60)
|
|||
|
|
print("🎯 重点观察:")
|
|||
|
|
print("1. 控制台是否显示完整的训练过程日志")
|
|||
|
|
print("2. 中文和表情符号是否正确显示")
|
|||
|
|
print("3. 训练进度是否实时输出")
|
|||
|
|
print("=" * 60)
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
# 先测试基本输出
|
|||
|
|
test_console_output()
|
|||
|
|
|
|||
|
|
# 再测试实际训练
|
|||
|
|
test_direct_trainer()
|