Artificial Intelligence, Machine Learning, Bert, Healthcare, AI.
of the model we just trained to make sure it has learned correctly. predictions, labels, _ = trainer.predict(test_tokenized_datasets) predictions = np.argmax(predictions, axis=2) true_predictions = [ [label_list[p] for (p, l) in zip(prediction, label) if l != -100] for prediction, label in