add bert script
This commit is contained in:
parent
e77c9e41d1
commit
1c4f054533
1
bert.py
1
bert.py
@ -1,5 +1,4 @@
|
|||||||
from transformers import AutoTokenizer, AutoModelForSequenceClassification, TrainingArguments, Trainer
|
from transformers import AutoTokenizer, AutoModelForSequenceClassification, TrainingArguments, Trainer
|
||||||
import random
|
|
||||||
import torch
|
import torch
|
||||||
|
|
||||||
PATHS = ['train/in.tsv', 'train/expected.tsv', 'dev-0/in.tsv', 'test-A/in.tsv', './dev-0/out.tsv', './test-A/out.tsv']
|
PATHS = ['train/in.tsv', 'train/expected.tsv', 'dev-0/in.tsv', 'test-A/in.tsv', './dev-0/out.tsv', './test-A/out.tsv']
|
||||||
|
Loading…
Reference in New Issue
Block a user