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

Add output

Browse Source
This commit is contained in:
nlitkowski 2021-06-22 22:58:14 +02:00
parent a9ba5d5728
commit 6e057483a7
3 changed files with 10429 additions and 3 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

8
main.py
View File

@ -47,7 +47,8 @@ def main(dirnames):
train_data = random.sample(train_data, 15000)
mname = PT_MODEL_NAME
if os.path.exists(MODEL_OUT_NAME):
pt = os.path.exists(MODEL_OUT_NAME)
if pt:
mname = MODEL_OUT_NAME
tokenizer = AutoTokenizer.from_pretrained(mname)
model = AutoModelForSequenceClassification.from_pretrained(
@ -69,8 +70,9 @@ def main(dirnames):
print("Starting training...")
trainer.train()
trainer.save_model(MODEL_OUT_NAME)
if not pt:
trainer.train()
trainer.save_model(MODEL_OUT_NAME)
print("Predicting outputs...")

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

File diff suppressed because it is too large Load Diff