Update DistilBERT.py

DistilBERT.py

@@ -17,7 +17,7 @@ TRAIN_BATCH_SIZE = 4
 VALID_BATCH_SIZE = 4
 EPOCHS = 1
 LEARNING_RATE = 1e-05
-tokenizer_DB = DistilBertTokenizer.from_pretrained('distilbert-base-uncased', truncation=True, do_lower_case=True)
+tokenizer = DistilBertTokenizer.from_pretrained('distilbert-base-uncased', truncation=True, do_lower_case=True)
 
 # Tạo dataframe
 train_df_DB = pd.read_csv('./data/train.csv')
@@ -30,7 +30,7 @@ test_df_DB['label'] = test_df_DB.iloc[:, 2:].values.tolist()
 class BinaryLabel(Dataset):
 
     def __init__(self, dataframe, tokenizer, max_len):
-        self.tokenizer = tokenizer_DB
+        self.tokenizer = tokenizer
         self.data = dataframe
         self.text = dataframe.text
         self.targets = self.data.label