Machine Learning, Deep Learning, Artificial Intelligence, AI.
compute_metrics(eval_pred): predictions = np.argmax(eval_pred.predictions, axis = 1) return metric.compute(predictions = predictions, references = eval_pred.label_ids) # Setting trainer trainer = Trainer( model=hubert_model, args = training_args, train_dataset = df['train'], eval_dataset =