README.md 1.08 KB
Newer Older
smanjil's avatar
smanjil committed
1
2
---
language: de
smanjil's avatar
smanjil committed
3
4
tags: 
- exbert
smanjil's avatar
smanjil committed
5
6
---

smanjil's avatar
smanjil committed
7
8
9
10
<a href="https://huggingface.co/exbert/?model=smanjil/German-MedBERT">
	<img width="300px" src="https://cdn-media.huggingface.co/exbert/button.png">
</a>

smanjil's avatar
smanjil committed
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
# German Medical BERT

This is a fine-tuned model on Medical domain for German language and based on German BERT.

## Overview
**Language model:** bert-base-german-cased

**Language:** German

**Fine-tuning:** Medical articles (diseases, symptoms, therapies, etc..)

**Eval data:** NTS-ICD-10 dataset (Classification)

**Infrastructure:** Gogle Colab


## Details
- We fine-tuned using Pytorch with Huggingface library on Colab GPU.
- With standard parameter settings for fine-tuning as mentioned in original BERT's paper.
- 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)

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

Get in touch:
[LinkedIn](https://www.linkedin.com/in/manjil-shrestha-038527b4/)