From 8f410ae8099a70c787061be641da662b7d0caf45 Mon Sep 17 00:00:00 2001
From: s440058 <pioula@st.amu.edu.pl>
Date: Mon, 21 Jun 2021 13:03:59 +0200
Subject: [PATCH] add roberta script

---
 bert.py => roberta.py | 2 --
 1 file changed, 2 deletions(-)
 rename bert.py => roberta.py (99%)

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])