加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
2_fit.py 673 Bytes
一键复制 编辑 原始数据 按行查看 历史
William Wang 提交于 2023-04-22 15:26 . init
# 0. 导入必要的库
# TODO
from util improt load,get,dump
# 1. 加载训练集和测试集
X_train, X_test, y_train, y_test = load("X_train, X_test, y_train, y_test", f'{get("Xy_root")}/Xy')
# 2. 使用LazyClassifier进行快速模型评估
print("开始评估所有的模型:")
# TODO
print(scores) # 打印不同模型的评估结果对比
# 3. 获取F1分数最高的模型
# 获取F1分数最高的模型名称
# TODO
print("F1分数最高的模型是: ", best_model_name)
# 根据模型名称,从模型字典中获取模型对象
best_model = # TODO
# 4. 序列化最佳模型
dump(best_model, "最好的F1分数的模型", f'{get("model_root")}/best_model')
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化