domischwimmbeck
/

bert-base-german-cased-own-data-ner

Token Classification

Generated from Trainer

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

domischwimmbeck commited on May 20, 2022

Commit

2d52f99

·

1 Parent(s): 576c8d4

update model card README.md

Files changed (1) hide show

README.md +12 -12

README.md CHANGED Viewed

@@ -19,11 +19,11 @@ should probably proofread and complete it, then remove this comment. -->
 This model is a fine-tuned version of [bert-base-german-cased](https://huggingface.co/bert-base-german-cased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0456
-- Precision: 0.7190
-- Recall: 0.85
-- F1: 0.7791
-- Accuracy: 0.9904
 ## Model description
@@ -42,7 +42,7 @@ More information needed
 ### Training hyperparameters
 The following hyperparameters were used during training:
-- learning_rate: 2e-05
 - train_batch_size: 32
 - eval_batch_size: 32
 - seed: 42
@@ -54,12 +54,12 @@ The following hyperparameters were used during training:
 | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1     | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
-| No log        | 0.8   | 32   | 0.0442          | 0.8066    | 0.7893 | 0.7978 | 0.9903   |
-| No log        | 1.6   | 64   | 0.0435          | 0.7337    | 0.8464 | 0.7861 | 0.9884   |
-| No log        | 2.4   | 96   | 0.0366          | 0.7702    | 0.85   | 0.8081 | 0.9909   |
-| No log        | 3.2   | 128  | 0.0389          | 0.7697    | 0.8357 | 0.8014 | 0.9907   |
-| No log        | 4.0   | 160  | 0.0377          | 0.7664    | 0.8321 | 0.7979 | 0.9911   |
-| No log        | 4.8   | 192  | 0.0456          | 0.7190    | 0.85   | 0.7791 | 0.9904   |
 ### Framework versions

 This model is a fine-tuned version of [bert-base-german-cased](https://huggingface.co/bert-base-german-cased) on an unknown dataset.
 It achieves the following results on the evaluation set:
+- Loss: 0.0535
+- Precision: 0.7134
+- Recall: 0.8536
+- F1: 0.7772
+- Accuracy: 0.9895
 ## Model description
 ### Training hyperparameters
 The following hyperparameters were used during training:
+- learning_rate: 5e-05
 - train_batch_size: 32
 - eval_batch_size: 32
 - seed: 42
 | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1     | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
+| No log        | 0.8   | 32   | 0.0308          | 0.7593    | 0.8    | 0.7791 | 0.9917   |
+| No log        | 1.6   | 64   | 0.0342          | 0.7756    | 0.8393 | 0.8062 | 0.9911   |
+| No log        | 2.4   | 96   | 0.0457          | 0.7764    | 0.8679 | 0.8196 | 0.9906   |
+| No log        | 3.2   | 128  | 0.0383          | 0.7524    | 0.8464 | 0.7966 | 0.9911   |
+| No log        | 4.0   | 160  | 0.0420          | 0.7539    | 0.8536 | 0.8007 | 0.9907   |
+| No log        | 4.8   | 192  | 0.0535          | 0.7134    | 0.8536 | 0.7772 | 0.9895   |
 ### Framework versions