thatdramebaazguy commited on
Commit
4903823
1 Parent(s): fc062f3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +60 -1
README.md CHANGED
@@ -1 +1,60 @@
1
- Model Card coming soon!
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ datasets:
3
+ - imdb
4
+ - cornell_movie_dialogue
5
+ - SQuAD
6
+
7
+ language:
8
+ - English
9
+
10
+ thumbnail:
11
+
12
+ tags:
13
+ - roberta
14
+ - roberta-base
15
+ - question-answering
16
+ - qa
17
+ - movies
18
+
19
+ license: cc-by-4.0
20
+
21
+ ---
22
+ # roberta-base + DAPT + Domain-Specific QA
23
+
24
+ Objective:
25
+ This is Roberta Base with Domain Adaptive Pretraining on Movie Corpora --> Then a changed head to do the SQuAD Task. This makes a QA model capable of answering questions in the movie domain.
26
+ https://huggingface.co/thatdramebaazguy/movie-roberta-base was used as the MovieRoberta.
27
+
28
+ ```
29
+ model_name = "thatdramebaazguy/movie-roberta-squad"
30
+ pipeline(model=model_name, tokenizer=model_name, revision="v1.0", task="question-answering")
31
+ ```
32
+
33
+ ## Overview
34
+ **Language model:** roberta-base
35
+ **Language:** English
36
+ **Downstream-task:** QA
37
+ **Training data:** imdb, polarity movie data, cornell_movie_dialogue, 25mlens movie names, SQuADv1
38
+ **Eval data:** MoviesQA (From https://github.com/ibm-aur-nlp/domain-specific-QA)
39
+ **Infrastructure**: 1x Tesla v100
40
+ **Code:** See [example](https://github.com/adityaarunsinghal/Domain-Adaptation/blob/master/scripts/shell_scripts/train_movieR_just_squadv1.sh)
41
+
42
+ ## Hyperparameters
43
+ ```
44
+ Num examples = 88567
45
+ Num Epochs = 10
46
+ Instantaneous batch size per device = 32
47
+ Total train batch size (w. parallel, distributed & accumulation) = 32
48
+
49
+ ```
50
+ ## Performance
51
+
52
+ ### Eval on MoviesQA
53
+ - eval_samples = 5032
54
+ - exact_match = 50.1391
55
+ - f1 = 64.8813
56
+
57
+ Github Repo:
58
+ - [Domain-Adaptation Project](https://github.com/adityaarunsinghal/Domain-Adaptation/)
59
+
60
+ ---