Skip to content

Commit

Permalink
replace performance table with markdown (#8565)
Browse files Browse the repository at this point in the history
* replace performance table with markdown

* Update model_cards/smanjil/German-MedBERT/README.md

Co-authored-by: Julien Chaumond <chaumond@gmail.com>
  • Loading branch information
smanjil and julien-c committed Nov 18, 2020
1 parent e7f77fc commit 6c8fad4
Showing 1 changed file with 8 additions and 2 deletions.
10 changes: 8 additions & 2 deletions model_cards/smanjil/German-MedBERT/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
language: de
tags:
- exbert
- German
---

<a href="https://huggingface.co/exbert/?model=smanjil/German-MedBERT">
Expand All @@ -10,7 +11,7 @@ tags:

# German Medical BERT

This is a fine-tuned model on Medical domain for German language and based on German BERT.
This is a fine-tuned model on Medical domain for German language and based on German BERT. This model has only been trained to improve on target task (Masked Language Model). It can later be used to perform a downstream task of your needs, while I performed it for NTS-ICD-10 text classification task.

## Overview
**Language model:** bert-base-german-cased
Expand All @@ -30,7 +31,12 @@ This is a fine-tuned model on Medical domain for German language and based on Ge
- Although had to train for upto 25 epochs for classification.

## Performance (Micro precision, recall and f1 score for multilabel code classification)
![performance](https://raw.githubusercontent.com/smanjil/finetune-lm/master/performance.png)

|Models |P |R |F1 |
|:-------------- |:------|:------|:------|
|German BERT |86.04 |75.82 |80.60 |
|German MedBERT-256 |87.41 |77.97 |82.42 |
|German MedBERT-512 |87.75 |78.26 |82.73 |

## Author
Manjil Shrestha: `shresthamanjil21 [at] gmail.com`
Expand Down

0 comments on commit 6c8fad4

Please sign in to comment.