torchdms.evaluation.build_evaluation_dict

torchdms.evaluation.build_evaluation_dict(model, test_data, device='cpu')

Evaluate & organize all testing data paired with metadata.

A function which takes a trained model, matching test dataset (BinaryMapDataset w/ the same input dimensions.) and return a dictionary containing the

• samples: binary encodings numpy array shape (num samples, num possible mutations)

• predictions and targets: both numpy arrays of shape (num samples, num targets)

This should have everything mostly needed to do plotting about testing data (not things like loss or latent space prediction).