src.ml.utils

Utility functions for machine learning operations.

Functions

calc_blur_acc_change	Calculate the difference in accuracy between min and max blur levels for all Stimulus models.
calc_model_frobenius_norm	Calculate the frobenius norm of model parameters.
calc_multi_acc	param y_pred A tensor of predicted values.
count_model_params	Count the number of parameters in a model.
generate_score_report	Generates a score report.
get_class_weights	Get balanced weights per class for cross-entropy loss.
get_latent_emb_per_class	Get latent embedding divided by output class labels.
plot_loss_acc_curves	Plot loss and accuracy curves.
print_log	Print training logs.
set_seed