add roberta script

This commit is contained in:
s440058 2021-06-21 13:03:59 +02:00
parent 1c4f054533
commit 8f410ae809

View File

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