nickprock commited on
Commit
c55afaf
·
1 Parent(s): 2efdb7a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -3
README.md CHANGED
@@ -35,18 +35,27 @@ It achieves the following results on the evaluation set:
35
 
36
  ## Model description
37
 
38
- More information needed
 
 
39
 
40
  ## Intended uses & limitations
41
 
42
- More information needed
43
 
44
  ## Training and evaluation data
45
 
46
- More information needed
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: