# 当我们训练好这个模型文件,怎么使用呢?
# 模型推理参考代码,你可以在本地XEdu一键安装包中运行:
from XEdu.hub import Workflow as wf # 使用BaseML训练的鸢尾花聚类模型推理
baseml = wf(task='baseml',checkpoint='./my_checkpoints/xxx/baseml.pkl') # 指定使用的pkl模型
data = [[5.1,1.5],[7,4.7]] # 推理的范例数据,你需要修改为和你训练时传入数据维度一致
result= baseml.inference(data=data) # 进行模型推理
format_output = baseml.format_output(lang='zh') # 推理结果格式化输出
模型训练已经完成,是否要将模型格式转换为onnx格式?
您的设置已成功保存。
训练已经结束,模型权重和日志已存储至……
模型转换中……请稍候
训练已终止,请重新训练