s440058/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

add roberta script

Browse Source
This commit is contained in:
s440058 2021-06-21 13:03:59 +02:00
parent 1c4f054533
commit 8f410ae809
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
bert.py → roberta.py
View File

@ -32,7 +32,6 @@ def prepare(data_train_X, data_train_Y):
return train_dataset, model return train_dataset, model
def trainer(train_dataset, model): def trainer(train_dataset, model):
training_args = TrainingArguments( training_args = TrainingArguments(
output_dir='./results', # output directory output_dir='./results', # output directory
@ -53,7 +52,6 @@ def trainer(train_dataset, model):
trainer.train() trainer.train()
def main(): def main():
#data #data
X_train = get_data(PATHS[0]) X_train = get_data(PATHS[0])