add roberta script

2021-06-21 13:03:59 +02:00 · 2021-06-21 13:03:59 +02:00 · 8f410ae809
commit 8f410ae809
parent 1c4f054533
1 changed files with 0 additions and 2 deletions
--- a/roberta.py
+++ b/roberta.py
@ -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])