hookeai.utilities.prediction_metrics.get_sample_n_features

get_sample_n_features(sample_results)

Get number of sample prediction features.

Parameters:

sample_results (dict) – Sample prediction results. Features predictions (and ground-truth if available) are stored as torch.Tensor(2d) of shape (sequence_length, n_features).

Returns:

n_features – Number of prediction features.

Return type:

int