trainer.evaluate
takes a dataset, so you can pass the test split if you want to evaluate on the test set.
Or there is trainer.predict
if you just want the predictions.
trainer.evaluate
takes a dataset, so you can pass the test split if you want to evaluate on the test set.
Or there is trainer.predict
if you just want the predictions.