随笔分类 -  模型训练相关问题

摘要:`KFold(n_splits=5, shuffle=True, random_state=3)` 阅读全文
posted @ 2020-11-25 15:27 hifalee 阅读(86) 评论(0) 推荐(0)
摘要:预处理excel文件中的数据不平衡问题 data = pd.read_excel ("file) list_label = [] train_list, dev_list, test_list = [],[],[] data_value = data.value for i in range(len 阅读全文
posted @ 2020-11-25 15:06 hifalee 阅读(154) 评论(0) 推荐(0)
摘要:shutil.rmtree(params.save) #删除文件保存相关文件 os.mkdir(params.save) #创建相关文件 torch.save(model, params.save+'/model.pkl') #以torch为例子,然后保存 def delete_file(path) 阅读全文
posted @ 2020-11-25 09:02 hifalee 阅读(165) 评论(0) 推荐(0)
摘要:f1 = metrics.f1_score(all_target, all_pred) 阅读全文
posted @ 2020-11-25 08:59 hifalee 阅读(423) 评论(0) 推荐(0)