s434704/paranormal-or-skeptic-ISI-public
1
0
Fork 0
forked from kubapok/paranormal-or-skeptic-ISI-public
Code Issues Pull Requests Projects Releases Wiki Activity

Final version classification with transformer

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5272
dev-0/out.tsv Normal file
View File

File diff suppressed because it is too large Load Diff

2
predict.py
View File

@ -31,7 +31,7 @@ class CustomDataset(torch.utils.data.Dataset):
tokenizer = BertTokenizer.from_pretrained("bert-base-uncased") 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) model = BertForSequenceClassification.from_pretrained(model_path, num_labels=2)
trainer = Trainer(model) trainer = Trainer(model)

5152
test-A/out.tsv Normal file
View File

File diff suppressed because it is too large Load Diff

289579
train/out.tsv Normal file
View File

File diff suppressed because it is too large Load Diff