src.ml.test.test_model

test_model(model, test_loader, device)

Testing loop.

model:

PyTorch model object.

test_loader:

Dataloader for test dataset.

device:

CUDA/CPU identifier.

Returns

A tuple of lists (y_true_list, y_pred_list).

Return type

tuple