Final version classification with transformer

This commit is contained in:
root 2021-06-22 01:31:56 +00:00
parent 6ad8ba24ba
commit cf933ff698
4 changed files with 300004 additions and 1 deletions

5272
dev-0/out.tsv Normal file

File diff suppressed because it is too large Load Diff

View File

@ -31,7 +31,7 @@ class CustomDataset(torch.utils.data.Dataset):
tokenizer = BertTokenizer.from_pretrained("bert-base-uncased")
model_path = "model/checkpoint-1500"
model_path = "model/checkpoint-18500"
model = BertForSequenceClassification.from_pretrained(model_path, num_labels=2)
trainer = Trainer(model)

5152
test-A/out.tsv Normal file

File diff suppressed because it is too large Load Diff

289579
train/out.tsv Normal file

File diff suppressed because it is too large Load Diff