{ "cells": [ { "cell_type": "markdown", "id": "4dd2d9a8", "metadata": {}, "source": [ "# bert-base-NER\n" ] }, { "cell_type": "markdown", "id": "39c0b4be", "metadata": {}, "source": [ "## Model description\n" ] }, { "cell_type": "markdown", "id": "0961b24f", "metadata": {}, "source": [ "**bert-base-NER** is a fine-tuned BERT model that is ready to use for **Named Entity Recognition** and achieves **state-of-the-art performance** for the NER task. It has been trained to recognize four types of entities: location (LOC), organizations (ORG), person (PER) and Miscellaneous (MISC).\n" ] }, { "cell_type": "markdown", "id": "58641459", "metadata": {}, "source": [ "Specifically, this model is a *bert-base-cased* model that was fine-tuned on the English version of the standard [CoNLL-2003 Named Entity Recognition](https://www.aclweb.org/anthology/W03-0419.pdf) dataset.\n" ] }, { "cell_type": "markdown", "id": "9da0ddda", "metadata": {}, "source": [ "If you'd like to use a larger BERT-large model fine-tuned on the same dataset, a **bert-large-NER** version is also available.\n" ] }, { "cell_type": "markdown", "id": "4d5adc68", "metadata": {}, "source": [ "## How to use\n" ] }, { "cell_type": "code", "execution_count": null, "id": "88f3ea49", "metadata": {}, "outputs": [], "source": [ "!pip install --upgrade paddlenlp" ] }, { "cell_type": "code", "execution_count": null, "id": "76ef1f0e", "metadata": {}, "outputs": [], "source": [ "import paddle\n", "from paddlenlp.transformers import AutoModel\n", "\n", "model = AutoModel.from_pretrained(\"dslim/bert-base-NER\")\n", "input_ids = paddle.randint(100, 200, shape=[1, 20])\n", "print(model(input_ids))" ] }, { "cell_type": "markdown", "id": "137e381c", "metadata": {}, "source": [ "## Citation\n", "```\n", "@inproceedings{tjong-kim-sang-de-meulder-2003-introduction,\n", "title = \"Introduction to the {C}o{NLL}-2003 Shared Task: Language-Independent Named Entity Recognition\",\n", "author = \"Tjong Kim Sang, Erik F. and\n", "De Meulder, Fien\",\n", "booktitle = \"Proceedings of the Seventh Conference on Natural Language Learning at {HLT}-{NAACL} 2003\",\n", "year = \"2003\",\n", "url = \"https://www.aclweb.org/anthology/W03-0419\",\n", "pages = \"142--147\",\n", "}\n", "```" ] }, { "cell_type": "markdown", "id": "3a632df2", "metadata": {}, "source": [ "> 此模型介绍及权重来源于[https://huggingface.co/dslim/bert-base-NER](https://huggingface.co/dslim/bert-base-NER),并转换为飞桨模型格式。\n" ] } ], "metadata": { "kernelspec": { "display_name": "Python 3 (ipykernel)", "language": "python", "name": "python3" }, "language_info": { "codemirror_mode": { "name": "ipython", "version": 3 }, "file_extension": ".py", "mimetype": "text/x-python", "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", "version": "3.7.13" } }, "nbformat": 4, "nbformat_minor": 5 }