covid19分类:使用LitCovid数据集和Hedwig库在COVID 19文献​​上测试文档分类 源码

COVID-19文件分类 此仓库提供了一个平台来测试COVID-19文献​​上的文档分类模型。 它是库的扩展,包含所有必要的代码,可在从集合创建的COVID-19数据集上重现某些文档分类模型的结果。 有关测试的模型和进行的实验的更多信息,请参见。 为了导入自定义模型,对Hedwig库进行了修改,使其可以与PyTorch和Transformers库的较新版本一起使用。 它还进行了扩展,以使能够使用模型。 数据 LitCovid数据 在以下目录下找到的LitCovid文档分类数据集可用于复制本文中找到的结果。 hedwig-data/datasets/LitCovid/ 你可以找到海维库兼容