Update training.py

training.py

@@ -21,8 +21,15 @@ data_collator = DataCollatorWithPadding(tokenizer=tokenizer, return_tensors="tf"
 
 tf_train_dataset = tokenized_dataset["train"].to_tf_dataset(
     columns=["attention_mask", "input_ids", "token_type_ids"],
-    label_cols=["labels"],
+    label_cols=["answer"],
     shuffle=True,
     collate_fn=data_collator,
     batch_size=8,
+)
+tf_validation_dataset = tokenized_datasets["validation"].to_tf_dataset(
+    columns=["attention_mask", "input_ids", "token_type_ids"],
+    label_cols=["answer"],
+    shuffle=False,
+    collate_fn=data_collator,
+    batch_size=8,
 )