syubraj/espanyol_bert_based_cased_ner

README.md

@@ -7,15 +7,6 @@ tags:
 model-index:
 - name: fine_tune_bert_output
   results: []
-datasets:
-- unimelb-nlp/wikiann
-language:
-- es
-metrics:
-- recall
-- precision
-- f1
-pipeline_tag: token-classification
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -34,75 +25,18 @@ It achieves the following results on the evaluation set:
 - Org F1: 0.8663
 - Per F1: 0.9367
 
-## Labels
-The following table represents the labels used by the model along with their corresponding indices:
-
-| Index | Label   |
-|-------|---------|
-| 0     | O       |
-| 1     | B-PER   |
-| 2     | I-PER   |
-| 3     | B-ORG   |
-| 4     | I-ORG   |
-| 5     | B-LOC   |
-| 6     | I-LOC   |
-
-### Label Descriptions
-- **O**: Outside of a named entity.
-- **B-PER**: Beginning of a person's name.
-- **I-PER**: Inside a person's name.
-- **B-ORG**: Beginning of an organization's name.
-- **I-ORG**: Inside an organization's name.
-- **B-LOC**: Beginning of a location name.
-- **I-LOC**: Inside a location name.
-
-## Inference Example
-```python
-from transformers import AutoModelForTokenClassification, AutoTokenizer, pipeline
-
-# Load the model and tokenizer
-model_name = "syubraj/espanyol_bert_based_ner"
-model = AutoModelForTokenClassification.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-
-# Custom label mapping
-custom_label_mapping = {
-    0: 'O',        # Outside of any named entity
-    1: 'B-PER',    # Beginning of a person's name
-    2: 'I-PER',    # Inside a person's name
-    3: 'B-ORG',    # Beginning of an organization's name
-    4: 'I-ORG',    # Inside an organization's name
-    5: 'B-LOC',    # Beginning of a location's name
-    6: 'I-LOC',    # Inside a location's name
-}
-
-# pipeline for Named Entity Recognition (NER)
-ner_pipeline = pipeline("ner", model=model, tokenizer=tokenizer)
-
-# Input text
-text = "Donald trabaja en Twitter"
-
-raw_results = ner_pipeline(text)
-
-results_with_labels = []
-for entity in raw_results:
-    label_index = int(entity['entity'].split('_')[1])
-    
-    entity_label = custom_label_mapping.get(label_index, "UNKNOWN")
-    
-    entity_with_label = {
-        "entity": entity_label,  
-        "word": entity["word"],
-        "start": entity["start"],
-        "end": entity["end"],
-        "score": entity["score"],
-    }
-    results_with_labels.append(entity_with_label)
-
-print("NER Results:")
-for result in results_with_labels:
-    print(result)
-```
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
 ## Training procedure
 
 ### Training hyperparameters
@@ -139,4 +73,4 @@ The following hyperparameters were used during training:
 - Transformers 4.44.2
 - Pytorch 2.4.1+cu121
 - Datasets 3.2.0
-- Tokenizers 0.19.1
+- Tokenizers 0.19.1