文件名称：pytorch_ner:使用PyTorch训练NER模型的管道
文件大小：38KB
文件格式：ZIP
更新时间：2024-06-13 03:09:20
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)