import fastdm
import pandas

%load_ext autoreload
%autoreload 2

data = pandas.load('/home/gdholla1/data/tdcs/data/leiden/leiden.pandas')

model = fastdm.FastDM(data, 
                      depends_on={'a':['acc_spd', 'stimulation'],
                                  't0':['acc_spd', 'stimulation'],},
                      method='ml')

model.fit(nproc=10)

pars = model.get_parameters()

pars.parameters

p = pars.melted_parameters('t0')
p

import seaborn as sns

p = pars.melted_parameters('t0')
sns.factorplot('acc_spd', 't0', hue='stimulation', data=p, units='subj_idx',)

p = pars.melted_parameters('a')
sns.factorplot('stimulation', 'a', hue='acc_spd', data=p, units='subj_idx',)