文件名称:pytorch-nli:pytorch实现snli和mnli任务的各种模型
文件大小:11KB
文件格式:ZIP
更新时间:2024-06-07 10:15:02
natural-language-processing pytorch natural-language-inference snli natural-language-understanding
pytorch-nli 该存储库旨在为自然语言推理(NLI)任务提供所有基线模型。 该存储库的主要目的是为处理NLI任务提供可扩展的代码结构和框架。 可以采用该存储库并将其用于训练模型和数据集的各种组合。 还可以对经过训练的模型和经过预先训练的模型进行评估。 该存储库是使用Pytorch和Torchtext编写的。 支持的选项 模型 数据集 双线性STM SNLI -- MultiNLI 结果 模型 验证 自测 BiLSTM 84.515 84.151 模型 multinli-dev-matched multinli-dev-mismatched BiLSTM 71.075 70.555 设置 conda: conda env create -f environment.yml # To download the english tokenizer from
【文件预览】:
pytorch-nli-master
----train.py(4KB)
----models()
--------bilstm.py(2KB)
--------__init__.py(21B)
----scripts()
--------download_pretrained.py(2KB)
----utils.py(2KB)
----requirements.txt(558B)
----datasets()
--------multinli.py(2KB)
--------__init__.py(43B)
--------snli.py(1KB)
----environment.yml(2KB)
----README.md(4KB)
----evaluate.py(3KB)
----.gitignore(91B)