Usage Sample
''''''''''''
.. code:: python
import torch
from transformers import AutoTokenizer
from nerx import NER, Collator
from model_wrapper import ClassifyModelWrapper
model = NER('hlf/rb3', num_classes=8)
tokenizer = AutoTokenizer.from_pretrained('hlf/rb3')
wrapper = ClassifyModelWrapper(model)
history = wrapper.train(train_set, val_set,
collate_fn=Collator(tokenizer, label_padding_id=7))
Raw data
{
"_id": null,
"home_page": "https://gitee.com/summry/NERX",
"name": "NERX",
"maintainer": null,
"docs_url": null,
"requires_python": ">=3.6",
"maintainer_email": null,
"keywords": "Pytorch, AI, Machine learning, Deep learning, torch, NER, Named Entity Recognition",
"author": "summy",
"author_email": "fkfkfk2024@2925.com",
"download_url": "https://files.pythonhosted.org/packages/56/1d/7e6b9613f9d2b8dcfed440abc1d9867352f5fbd6139b3f1567f31c8a76c4/NERX-0.0.6.tar.gz",
"platform": null,
"description": "Usage Sample\n''''''''''''\n\n.. code:: python\n\n import torch\n from transformers import AutoTokenizer\n from nerx import NER, Collator\n from model_wrapper import ClassifyModelWrapper\n \n model = NER('hlf/rb3', num_classes=8)\n tokenizer = AutoTokenizer.from_pretrained('hlf/rb3')\n wrapper = ClassifyModelWrapper(model)\n history = wrapper.train(train_set, val_set, \n collate_fn=Collator(tokenizer, label_padding_id=7))\n\n\n",
"bugtrack_url": null,
"license": null,
"summary": "Named Entity Recognition",
"version": "0.0.6",
"project_urls": {
"Homepage": "https://gitee.com/summry/NERX"
},
"split_keywords": [
"pytorch",
" ai",
" machine learning",
" deep learning",
" torch",
" ner",
" named entity recognition"
],
"urls": [
{
"comment_text": "",
"digests": {
"blake2b_256": "561d7e6b9613f9d2b8dcfed440abc1d9867352f5fbd6139b3f1567f31c8a76c4",
"md5": "cdc2293824dc506dabdb5b6a756367d0",
"sha256": "5af4062a4169522b1b01d7fd5f1b282638586a2dffb845e46b7b908139a3b521"
},
"downloads": -1,
"filename": "NERX-0.0.6.tar.gz",
"has_sig": false,
"md5_digest": "cdc2293824dc506dabdb5b6a756367d0",
"packagetype": "sdist",
"python_version": "source",
"requires_python": ">=3.6",
"size": 7192,
"upload_time": "2025-03-15T05:06:04",
"upload_time_iso_8601": "2025-03-15T05:06:04.575936Z",
"url": "https://files.pythonhosted.org/packages/56/1d/7e6b9613f9d2b8dcfed440abc1d9867352f5fbd6139b3f1567f31c8a76c4/NERX-0.0.6.tar.gz",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2025-03-15 05:06:04",
"github": false,
"gitlab": false,
"bitbucket": false,
"codeberg": false,
"lcname": "nerx"
}