get_r2s_across_trials¶

behavenet.plotting.decoder_utils.get_r2s_across_trials(hparams, best_models_df)[source]¶

Calculate R^2 across all test trials (rather than on a trial-by-trial basis)

Parameters:

hparams (dict) – needs to contain the keys ‘lab’, ‘experiment’, ‘animal’, ‘session’, ‘model_type’, ‘region’, ‘n_hid_layers’, ‘n_lags’
best_models_df (pd.DataFrame) – output of get_best_models()

Returns:

test R^2 across all trials

Return type:

pd.DataFrame