in src/train_ner.py [0:0]
def run(self, tokenized_dataset):
trainer = Trainer(
model=self.model,
args=self.training_args, # Hyperparamaters
train_dataset=tokenized_dataset["train"], # training data
eval_dataset=tokenized_dataset["validation"], # validation data
tokenizer=self.tokenizer, # tokenizer
compute_metrics=self.compute_metrics, # model perfomance evaluation metric
)
trainer.train()