Update README.md
Browse files
README.md
CHANGED
@@ -35,18 +35,27 @@ It achieves the following results on the evaluation set:
|
|
35 |
|
36 |
## Model description
|
37 |
|
38 |
-
|
|
|
|
|
39 |
|
40 |
## Intended uses & limitations
|
41 |
|
42 |
-
|
43 |
|
44 |
## Training and evaluation data
|
45 |
|
46 |
-
|
47 |
|
48 |
## Training procedure
|
49 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
50 |
### Training hyperparameters
|
51 |
|
52 |
The following hyperparameters were used during training:
|
|
|
35 |
|
36 |
## Model description
|
37 |
|
38 |
+
This is my first fine-tuning experiment using Hugging Face.
|
39 |
+
Using distilBERT as a pretrained model, I trained a classifier for online banking queries.
|
40 |
+
It could be useful for addressing tickets.
|
41 |
|
42 |
## Intended uses & limitations
|
43 |
|
44 |
+
The model can be used on text classification. In particular is fine tuned on banking domain.
|
45 |
|
46 |
## Training and evaluation data
|
47 |
|
48 |
+
The dataset used is [banking77](https://huggingface.co/datasets/banking77)
|
49 |
|
50 |
## Training procedure
|
51 |
|
52 |
+
```
|
53 |
+
from transformers import pipeline
|
54 |
+
|
55 |
+
pipe = pipeline("text-classification", model="nickprock/distilbert-base-uncased-banking77-classification")
|
56 |
+
pipe("I can't pay by my credit card")
|
57 |
+
```
|
58 |
+
|
59 |
### Training hyperparameters
|
60 |
|
61 |
The following hyperparameters were used during training:
|