33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from function.model_manager import ModelManager
|
|
|
|
# 创建模型管理器实例
|
|
manager = ModelManager()
|
|
|
|
# 获取所有已训练模型
|
|
result = manager.get_finished_models(
|
|
page=1,
|
|
page_size=10,
|
|
experiment_name='breast_cancer_classification_2'
|
|
)
|
|
|
|
# 打印结果
|
|
print("\n已训练模型列表:")
|
|
print(f"状态: {result['status']}")
|
|
if result['status'] == 'success':
|
|
print(f"\n总数: {result['total_count']}")
|
|
print(f"当前页: {result['page']}")
|
|
print(f"每页数量: {result['page_size']}")
|
|
print("\n模型列表:")
|
|
for model in result['models']:
|
|
print(f"\n模型ID: {model['model_id']}")
|
|
print(f"算法: {model['algorithm']}")
|
|
print(f"任务类型: {model['task_type']}")
|
|
print(f"数据集: {model['dataset']}")
|
|
|
|
print(f"训练开始时间: {model['training_start_time']}")
|
|
print(f"训练结束时间: {model['training_end_time']}")
|
|
print("评估指标:")
|
|
for metric_name, metric_value in model['metrics'].items():
|
|
print(f" {metric_name}: {metric_value:.4f}")
|
|
else:
|
|
print(f"错误信息: {result['message']}") |