model update
Browse files
README.md
CHANGED
@@ -21,7 +21,7 @@ widget:
|
|
21 |
- text: "Beyonce further expanded her acting career, starring as blues singer Etta James in the 2008 musical biopic, <hl> Cadillac Records <hl> ."
|
22 |
example_title: "Question Generation Example 3"
|
23 |
model-index:
|
24 |
-
- name: lmqg/bart-base-squadshifts-reddit
|
25 |
results:
|
26 |
- task:
|
27 |
name: Text2text Generation
|
@@ -48,9 +48,9 @@ model-index:
|
|
48 |
value: 62.35
|
49 |
---
|
50 |
|
51 |
-
# Model Card of `lmqg/bart-base-squadshifts-reddit`
|
52 |
This model is fine-tuned version of [lmqg/bart-base-squad](https://huggingface.co/lmqg/bart-base-squad) for question generation task on the [lmqg/qg_squadshifts](https://huggingface.co/datasets/lmqg/qg_squadshifts) (dataset_name: reddit) via [`lmqg`](https://github.com/asahi417/lm-question-generation).
|
53 |
-
|
54 |
|
55 |
### Overview
|
56 |
- **Language model:** [lmqg/bart-base-squad](https://huggingface.co/lmqg/bart-base-squad)
|
@@ -66,7 +66,7 @@ This model is continuously fine-tuned with [lmqg/bart-base-squad](https://huggin
|
|
66 |
from lmqg import TransformersQG
|
67 |
|
68 |
# initialize model
|
69 |
-
model = TransformersQG(language="en", model="lmqg/bart-base-squadshifts-reddit")
|
70 |
|
71 |
# model prediction
|
72 |
questions = model.generate_q(list_context="William Turner was an English painter who specialised in watercolour landscapes", list_answer="William Turner")
|
@@ -77,7 +77,7 @@ questions = model.generate_q(list_context="William Turner was an English painter
|
|
77 |
```python
|
78 |
from transformers import pipeline
|
79 |
|
80 |
-
pipe = pipeline("text2text-generation", "lmqg/bart-base-squadshifts-reddit")
|
81 |
output = pipe("<hl> Beyonce <hl> further expanded her acting career, starring as blues singer Etta James in the 2008 musical biopic, Cadillac Records.")
|
82 |
|
83 |
```
|
@@ -85,7 +85,7 @@ output = pipe("<hl> Beyonce <hl> further expanded her acting career, starring as
|
|
85 |
## Evaluation
|
86 |
|
87 |
|
88 |
-
- ***Metric (Question Generation)***: [raw metric file](https://huggingface.co/lmqg/bart-base-squadshifts-reddit/raw/main/eval/metric.first.sentence.paragraph_answer.question.lmqg_qg_squadshifts.reddit.json)
|
89 |
|
90 |
| | Score | Type | Dataset |
|
91 |
|:-----------|--------:|:-------|:---------------------------------------------------------------------------|
|
@@ -119,7 +119,7 @@ The following hyperparameters were used during fine-tuning:
|
|
119 |
- gradient_accumulation_steps: 16
|
120 |
- label_smoothing: 0.15
|
121 |
|
122 |
-
The full configuration can be found at [fine-tuning config file](https://huggingface.co/lmqg/bart-base-squadshifts-reddit/raw/main/trainer_config.json).
|
123 |
|
124 |
## Citation
|
125 |
```
|
|
|
21 |
- text: "Beyonce further expanded her acting career, starring as blues singer Etta James in the 2008 musical biopic, <hl> Cadillac Records <hl> ."
|
22 |
example_title: "Question Generation Example 3"
|
23 |
model-index:
|
24 |
+
- name: lmqg/bart-base-squadshifts-reddit-qg
|
25 |
results:
|
26 |
- task:
|
27 |
name: Text2text Generation
|
|
|
48 |
value: 62.35
|
49 |
---
|
50 |
|
51 |
+
# Model Card of `lmqg/bart-base-squadshifts-reddit-qg`
|
52 |
This model is fine-tuned version of [lmqg/bart-base-squad](https://huggingface.co/lmqg/bart-base-squad) for question generation task on the [lmqg/qg_squadshifts](https://huggingface.co/datasets/lmqg/qg_squadshifts) (dataset_name: reddit) via [`lmqg`](https://github.com/asahi417/lm-question-generation).
|
53 |
+
|
54 |
|
55 |
### Overview
|
56 |
- **Language model:** [lmqg/bart-base-squad](https://huggingface.co/lmqg/bart-base-squad)
|
|
|
66 |
from lmqg import TransformersQG
|
67 |
|
68 |
# initialize model
|
69 |
+
model = TransformersQG(language="en", model="lmqg/bart-base-squadshifts-reddit-qg")
|
70 |
|
71 |
# model prediction
|
72 |
questions = model.generate_q(list_context="William Turner was an English painter who specialised in watercolour landscapes", list_answer="William Turner")
|
|
|
77 |
```python
|
78 |
from transformers import pipeline
|
79 |
|
80 |
+
pipe = pipeline("text2text-generation", "lmqg/bart-base-squadshifts-reddit-qg")
|
81 |
output = pipe("<hl> Beyonce <hl> further expanded her acting career, starring as blues singer Etta James in the 2008 musical biopic, Cadillac Records.")
|
82 |
|
83 |
```
|
|
|
85 |
## Evaluation
|
86 |
|
87 |
|
88 |
+
- ***Metric (Question Generation)***: [raw metric file](https://huggingface.co/lmqg/bart-base-squadshifts-reddit-qg/raw/main/eval/metric.first.sentence.paragraph_answer.question.lmqg_qg_squadshifts.reddit.json)
|
89 |
|
90 |
| | Score | Type | Dataset |
|
91 |
|:-----------|--------:|:-------|:---------------------------------------------------------------------------|
|
|
|
119 |
- gradient_accumulation_steps: 16
|
120 |
- label_smoothing: 0.15
|
121 |
|
122 |
+
The full configuration can be found at [fine-tuning config file](https://huggingface.co/lmqg/bart-base-squadshifts-reddit-qg/raw/main/trainer_config.json).
|
123 |
|
124 |
## Citation
|
125 |
```
|