thatdramebaazguy
commited on
Commit
•
54fc4b7
1
Parent(s):
1e9185c
update
Browse files
README.md
CHANGED
@@ -1,8 +1,4 @@
|
|
1 |
---
|
2 |
-
objective:
|
3 |
-
This is Roberta Base with Domain Adaptive Pretraining on Movie Corpora --> Then trained for the NER task using MIT Movie Dataset --> Then a changed head to do the SQuAD Task. This makes a QA model capable of answering questions in the movie domain, with additional information coming from a different task (NER - Task Transfer).
|
4 |
-
https://huggingface.co/thatdramebaazguy/movie-roberta-base was used as the MovieRoberta.
|
5 |
-
|
6 |
datasets:
|
7 |
- imdb (Movie corpus for Domain Adaptive Pretraining)
|
8 |
- cornell_movie_dialogue
|
@@ -25,6 +21,10 @@ license: cc-by-4.0
|
|
25 |
---
|
26 |
# roberta-base for QA
|
27 |
|
|
|
|
|
|
|
|
|
28 |
```
|
29 |
model_name = "thatdramebaazguy/movie-roberta-MITmovie-squad"
|
30 |
pipeline(model=model_name, tokenizer=model_name, revision="v1.0", task="question-answering")
|
|
|
1 |
---
|
|
|
|
|
|
|
|
|
2 |
datasets:
|
3 |
- imdb (Movie corpus for Domain Adaptive Pretraining)
|
4 |
- cornell_movie_dialogue
|
|
|
21 |
---
|
22 |
# roberta-base for QA
|
23 |
|
24 |
+
objective:
|
25 |
+
This is Roberta Base with Domain Adaptive Pretraining on Movie Corpora --> Then trained for the NER task using MIT Movie Dataset --> Then a changed head to do the SQuAD Task. This makes a QA model capable of answering questions in the movie domain, with additional information coming from a different task (NER - Task Transfer).
|
26 |
+
https://huggingface.co/thatdramebaazguy/movie-roberta-base was used as the MovieRoberta.
|
27 |
+
|
28 |
```
|
29 |
model_name = "thatdramebaazguy/movie-roberta-MITmovie-squad"
|
30 |
pipeline(model=model_name, tokenizer=model_name, revision="v1.0", task="question-answering")
|