Update README.md

README.md

@@ -7,16 +7,19 @@ model-index:
   results: []
 ---
 
-<!-- This model card has been generated automatically according to the information Keras had access to. You should
-probably proofread and complete it, then remove this comment. -->
 
 # jjglilleberg/bert-finetuned-ner-nbci-disease
 
-This model is a fine-tuned version of [bert-base-cased](https://huggingface.co/bert-base-cased) on an unknown dataset.
+This model is a fine-tuned version of [bert-base-cased](https://huggingface.co/bert-base-cased) on the [NCBI Disease Dataset](https://www.ncbi.nlm.nih.gov/research/bionlp/Data/disease/).
 It achieves the following results on the evaluation set:
-- Train Loss: 0.0211
-- Validation Loss: 0.0645
-- Epoch: 2
+- Precision: 0.759090909090909,
+- Recall: 0.8487928843710292,
+- F1: 0.8014397120575885,
+- Number: 787,
+- Overall_precision: 0.759090909090909,
+- Overall_recall: 0.8487928843710292,
+- Overall_f1: 0.8014397120575885,
+- Overall_accuracy: 0.9824785260799204
 
 ## Model description
 
@@ -24,18 +27,30 @@ More information needed
 
 ## Intended uses & limitations
 
-More information needed
-
-## Training and evaluation data
-
-More information needed
+The intended use of this model is for Disease Name Recognition and Concept Normalization.
 
 ## Training procedure
 
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- optimizer: {'name': 'AdamWeightDecay', 'learning_rate': {'class_name': 'PolynomialDecay', 'config': {'initial_learning_rate': 2e-05, 'decay_steps': 1020, 'end_learning_rate': 0.0, 'power': 1.0, 'cycle': False, 'name': None}}, 'decay': 0.0, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-08, 'amsgrad': False, 'weight_decay_rate': 0.01}
+- optimizer:
+  - 'name': 'AdamWeightDecay',
+  - 'learning_rate':
+    - 'class_name': 'PolynomialDecay',
+    - 'config':
+      - 'initial_learning_rate': 2e-05,
+      - 'decay_steps': 1020,
+      - 'end_learning_rate': 0.0,
+      - 'power': 1.0,
+      - 'cycle': False,
+      - 'name': None
+  - 'decay': 0.0,
+  - 'beta_1': 0.9,
+  - 'beta_2': 0.999,
+  - 'epsilon': 1e-08,
+  - 'amsgrad': False,
+  - 'weight_decay_rate': 0.01
 - training_precision: mixed_float16
 
 ### Training results