Update training_bert.py

training_bert.py

@@ -6,6 +6,9 @@ import torch
 from sklearn.model_selection import GroupShuffleSplit
 from transformers import AutoTokenizer,AutoModelForSequenceClassification
 from transformers import TrainingArguments
+from sklearn.metrics import mean_absolute_error
+from sklearn.metrics import mean_squared_error
+from sklearn.metrics import r2_score
 
 def remove_links(review):
     pattern = r'\bhttps?://\S+'
@@ -67,9 +70,7 @@ def preprocess_function_regr(examples):
 ds_train_regr_tok = ds_train_regr.map(preprocess_function_regr, remove_columns = ['Text'])
 ds_val_regr_tok = ds_val_regr.map(preprocess_function_regr, remove_columns = ['Text'])
 
-from sklearn.metrics import mean_absolute_error
-from sklearn.metrics import mean_squared_error
-from sklearn.metrics import r2_score
+
 
 def compute_metrics_for_regression(eval_pred):
     logits, labels = eval_pred