【文件属性】：
文件名称：pytorch_ner:使用PyTorch训练NER模型的管道
文件大小：38KB
文件格式：ZIP
更新时间：2021-05-24 09:22:40
nlp natural-language-processing deep-learning pipeline text
使用PyTorch命名实体识别（NER）
关于
使用PyTorch训练NER模型的管道。 支持ONNX导出。
用法
首先，安装依赖项：
# clone repo
git clone https://github.com/dayyass/pytorch_ner.git
# install dependencies
cd pytorch_ner
pip install -r requirements.txt
用户界面仅包含一个文件config.yaml 。 更改config.yaml以创建所需的配置，并使用以下命令启动管道：
python main.py --config config.yaml
如果未指定--config参数，则使用config.yaml。
要将训练后的模型导出到ONNX，请使用config.yaml：
save:
export_onnx: Tr
【文件预览】：
pytorch_ner-main
----Dockerfile(240B)
----.gitignore(53B)
----pytorch_ner()
--------prepare_data.py(2KB)
--------save.py(1KB)
--------onnx.py(3KB)
--------dataset.py(3KB)
--------train.py(5KB)
--------metrics.py(615B)
--------utils.py(807B)
--------nn_modules()
----README.md(2KB)
----.github()
--------workflows()
----.pre-commit-config.yaml(766B)
----tests()
--------test_train.py(2KB)
--------test_prepare_data.py(4KB)
--------test_onnx.py(276B)
--------test_dataset.py(3KB)
--------test_save.py(1012B)
--------__init__.py(0B)
--------test_nn_modules()
--------data()
----LICENSE(11KB)
----.dockerignore(20B)
----main.py(7KB)
----requirements.txt(526B)
----config.yaml(839B)