sklearn remove

src/run.py

@@ -2,15 +2,15 @@ from glob import glob
 from tqdm import tqdm
 import numpy as np
 import pickle
-from sklearn.model_selection import train_test_split
+#from sklearn.model_selection import train_test_split
 import torch
 import os
 import ast
-from sklearn.metrics import accuracy_score, recall_score, precision_score, f1_score
+#from sklearn.metrics import accuracy_score, recall_score, precision_score, f1_score
 from transformers import EarlyStoppingCallback
 from transformers import AutoConfig, AutoModel, AutoTokenizer, AutoModelForSequenceClassification
 from torch.utils.data import Dataset, DataLoader, RandomSampler, SequentialSampler
-from sklearn.utils import shuffle
+#from sklearn.utils import shuffle
 from transformers import get_cosine_schedule_with_warmup
 from torch.nn import functional as F
 import random