diff --git a/bert.py b/roberta.py similarity index 99% rename from bert.py rename to roberta.py index 8d1f0e3..610abf7 100644 --- a/bert.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])