Update README.md
Browse files
README.md
CHANGED
|
@@ -1,30 +1,3 @@
|
|
| 1 |
-
# Roberta-Base fine-tuned on [PubMed](https://pubmed.ncbi.nlm.nih.gov/) Abstract
|
| 2 |
-
> We limit the training textual data to the following [MeSH](https://www.ncbi.nlm.nih.gov/mesh/)
|
| 3 |
-
* All the child MeSH of ```Biomarkers, Tumor(D014408)```, including things like ```Carcinoembryonic Antigen(D002272)```
|
| 4 |
-
* All the child MeSH of ```Carcinoma(D002277)```, including things like all kinds of carcinoma: like ```Carcinoma, Lewis Lung(D018827)``` etc. around 80 kinds of carcinoma
|
| 5 |
-
* All the child MeSH of ```Clinical Trial(D016439)```
|
| 6 |
-
* The training text file amounts to 531Mb
|
| 7 |
-
## Training
|
| 8 |
-
* Trained on language modeling task, with ```mlm_probability=0.15```, on 2 Tesla V100 32G
|
| 9 |
-
```python
|
| 10 |
-
training_args = TrainingArguments(
|
| 11 |
-
output_dir=config.save, #select model path for checkpoint
|
| 12 |
-
overwrite_output_dir=True,
|
| 13 |
-
num_train_epochs=3,
|
| 14 |
-
per_device_train_batch_size=30,
|
| 15 |
-
per_device_eval_batch_size=60,
|
| 16 |
-
evaluation_strategy= 'steps',
|
| 17 |
-
save_total_limit=2,
|
| 18 |
-
eval_steps=250,
|
| 19 |
-
metric_for_best_model='eval_loss',
|
| 20 |
-
greater_is_better=False,
|
| 21 |
-
load_best_model_at_end =True,
|
| 22 |
-
prediction_loss_only=True,
|
| 23 |
-
report_to = "none")
|
| 24 |
-
```
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
| 28 |
---
|
| 29 |
language:
|
| 30 |
- en
|
|
@@ -36,5 +9,7 @@ tags:
|
|
| 36 |
license: apache-2.0
|
| 37 |
datasets:
|
| 38 |
- pubmed
|
|
|
|
|
|
|
| 39 |
---
|
| 40 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
language:
|
| 3 |
- en
|
|
|
|
| 9 |
license: apache-2.0
|
| 10 |
datasets:
|
| 11 |
- pubmed
|
| 12 |
+
widget:
|
| 13 |
+
- text: "The <mask> effects of hyperatomarin"
|
| 14 |
---
|
| 15 |
|