def figure_S1(multiprocessing=False):
pl.reversal_analysis(sessions_ntr, fig_no='S1A')
pl.stay_probability_analysis(sessions_ntr, fig_no='S1B')
if multiprocessing:
pp.enable_multiprocessing()
lr.logistic_regression(sessions, 'standard', fig_no='S1C')
lr.logistic_regression(sessions, 'lagged', fig_no='S1D')
def JAWS_LR_permutation_test(multiprocessing=False, n_perms=5000):
'''n_core : number of cores to use for parallel processing
n_perms: number of permutations to run.'''
if multiprocessing:
pp.enable_multiprocessing()
sa.logistic_regression_test(sessions_jaws, fig_no=0, n_perms=n_perms,
file_name='JAWS_LR_full')
def permuation_tests(multiprocessing=False):
gc.reversal_test(sessions, sessions_ntr, 'cross_subject')
if multiprocessing:
pp.enable_multiprocessing()
gc.model_fit_test(sessions,
sessions_ntr,
LR1_model,
'cross_subject',
file_name='LR1_ntr_test')
gc.model_fit_test(sessions,
sessions_ntr,
LR_lag_model,
'cross_subject',
file_name='LR_lag_ntr_test')
def LR_interaction_perm_test(multiprocessing=False, n_perms=5000):
if multiprocessing:
pp.enable_multiprocessing()
sa.log_reg_stim_x_group_interaction(sessions_jaws, sessions_ctrl,
n_perms=n_perms, file_name='LR_interaction_test')
def CTRL_LR_permutation_test(multiprocessing=False, n_perms=5000):
if multiprocessing:
pp.enable_multiprocessing()
sa.logistic_regression_test(sessions_ctrl, fig_no=0, n_perms=n_perms,
file_name='CTRL_LR_full')
def figure_6(multiprocessing=False):
if multiprocessing:
pp.enable_multiprocessing()
sa.logistic_regression_comparison(sessions_jaws, title='JAWS', fig_no='6D1')
sa.logistic_regression_comparison(sessions_ctrl, title='CTRL', fig_no='6D2')
sa.RL_LR_correlation(sessions_jaws, fig_no='6E')