CoolFace
Modelpublic

projecte-aina/multiner_ceil

sourceHugging Faceapache-2.0updated 2y agoView on Hugging Face
1likes41downloads
Model Card

Catalan BERTa (RoBERTa-large) finetuned for Named Entity Recognition.

Table of Contents

<details> <summary>Click to expand</summary>

Model description

The multiner is a Named Entity Recognition (NER) model for the Catalan language fine-tuned from the [BERTa] model, a RoBERTa base model pre-trained on a medium-size corpus collected from publicly available corpora and crawlers (check the BERTa model card for more details).

It has been trained with a dataset that contains 9 main types and 52 subtypes on all kinds of short texts, with almost 59K documents.

Intended uses and limitations

How to use

from transformers import pipeline

pipe = pipeline("ner", model="projecte-aina/multiner_ceil")
example = "George Smith Patton fué un general del Ejército de los Estados Unidos en Europa durante la Segunda Guerra Mundial. "

ner_entity_results = pipe(example, aggregation_strategy="simple")
print(ner_entity_results)

Limitations and bias

At the time of submission, no measures have been taken to estimate the bias embedded in the model. However, we are well aware that our models may be biased since the corpora have been collected using crawling techniques on multiple web sources. We intend to conduct research in these areas in the future, and if completed, this model card will be updated.

Training

We used the NERC dataset in Catalan called Catalan Entity Identification and Linking for training and evaluation.

Evaluation

Accuracy was calculated using the development set, and reflects the non-balanced nature of the dataset.

Major types

TypeAccuracynum. Instances in dev set
CW *0.8424551
GPE0.91419751
Other0.692824
building0.7362188
event0.7393000
location0.8193408
organization0.89517285
person0.90321689
product0.641038

*: Cultural Work

Subtypes

TypeAccuracynum. Instances in dev set
CW-broadcastprogram0.854765
CW-film0.809549
CW-music0.8621027
CW-other0.495555
CW-painting0.654205
CW-writtenart0.8141450
GPE0.91419751
Other0.692824
building-airport0.733176
building-governmentfacility0.51472
building-hospital0.805113
building-hotel0.68832
building-other0.7261585
building-religious0.01
building-restaurant0.45848
building-shops0.20634
building-sportsfacility0.74127
event-attack/terrorism/militaryconflict0.866411
event-disaster0.26123
event-other0.6951069
event-political0.527444
event-protest0.20729
event-sportsevent0.8221024
location-bodiesofwater0.865673
location-island0.457140
location-mountain0.781515
location-other0.7571602
location-park0.58193
location-road/railway/highway/transit0.805385
organization-education0.8682097
organization-government0.9052939
organization-media0.8881963
organization-onlinebusiness0.538197
organization-other0.7884733
organization-politicalparty0.9562272
organization-privatecompany0.8491809
organization-religious0.638210
organization-sportsteam0.9461065
person-actor/director0.7971480
person-artist/author0.8535812
person-athlete0.8711306
person-group0.485699
person-influencer0.017
person-other0.8118444
person-politician0.8633259
person-scholar/scientist0.728672
product-E-device0.51102
product-clothing0.22227
product-consumer_good0.020
product-food0.673324
product-other0.069
product-software0.67382
product-vehicle0.825114

Additional information

Author

Language Technologies Unit (LangTech) at the Barcelona Supercomputing Center (langtech@bsc.es)

Contact information

For further information, send an email to langtech@bsc.es

Copyright

Copyright (c) 2023 Language Technologies Unit (LangTech) at Barcelona Supercomputing Center

Licensing Information

Apache License, Version 2.0

Funding

This work/research has been promoted and financed by the Government of Catalonia through the Aina project.

Citation information

Disclaimer

<details> <summary>Click to expand</summary>

The models published in this repository are intended for a generalist purpose and are available to third parties. These models may have bias and/or any other undesirable distortions.

When third parties, deploy or provide systems and/or services to other parties using any of these models (or using systems based on these models) or become users of the models, they should note that it is their responsibility to mitigate the risks arising from their use and, in any event, to comply with applicable regulations, including regulations regarding the use of Artificial Intelligence.

In no event shall the owner and creator of the models (BSC – Barcelona Supercomputing Center) be liable for any results arising from the use made by third parties of these models.