s478855/paranormal-or-skeptic
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

s478855 - correct outs

Browse Source
This commit is contained in:
ulaniuk 2022-06-22 13:32:19 +02:00
parent 67f4a46156
commit bd06f0ed0c
4 changed files with 3670 additions and 3700 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1460
dev-0/out.tsv
View File

File diff suppressed because it is too large Load Diff

2305
run2.ipynb
View File

File diff suppressed because it is too large Load Diff

13
run2.py
View File

@ -138,8 +138,14 @@ testA_dataset = testA_dataset.map(tokenize, batched=True, batch_size=len(train_d
dev_dataset.set_format('torch', columns=['input_ids', 'attention_mask', 'label'])
testA_dataset.set_format('torch', columns=['input_ids', 'attention_mask', 'label'])
y_pred_dev = trainer.predict(dev_dataset).label_ids
y_pred_test = trainer.predict(testA_dataset).label_ids
y_pred_dev = trainer.predict(dev_dataset).predictions
y_pred_test = trainer.predict(testA_dataset).predictions
def get_labels(predictions):
return [0 if a > b else 1 for a, b in predictions]
y_pred_dev = get_labels(y_pred_dev)
y_pred_test = get_labels(y_pred_test)
with open('/content/drive/MyDrive/eks/dev-0/out.tsv', 'wt') as f:
for pred in y_pred_dev:
@ -147,4 +153,5 @@ with open('/content/drive/MyDrive/eks/dev-0/out.tsv', 'wt') as f:
with open('/content/drive/MyDrive/eks/test-A/out.tsv', 'wt') as f:
for pred in y_pred_test:
f.write(str(pred)+'\n')
f.write(str(pred)+'\n')

3592
test-A/out.tsv
View File

File diff suppressed because it is too large Load Diff