diff --git a/py/20200116_efo_analysis.py b/py/20200116_efo_analysis.py index 4a65c45..6e1ee0f 100644 --- a/py/20200116_efo_analysis.py +++ b/py/20200116_efo_analysis.py @@ -100,9 +100,11 @@ #%% TODO: model 3 # ----------------------------------------------- -sd_constructs = df.groupby('construct').std().construct_hits +ef_construct_mean = df.groupby('context').std().construct_hits['EF'] - +display(ef_construct_mean) +df[df.context=='EF'].groupby('construct').sum() +#%% with pm.Model() as model3: context_mu = pm.Normal('context_mu', mu=hits_grand_mean, sigma=sd_contexts*100, shape=n_context) context_sigma = pm.Uniform('context_sigma', sd_contexts/1000, sd_contexts*1000, shape=n_context) @@ -122,9 +124,8 @@ trace = pm.sample(model=model3) #pm.traceplot(trace, var_names=['construct_mu','constructs_sigma','constructs diff (EXAMPLE)']) - pm.plot_posterior(trace, - var_names=['construct_mu','constructs_sigma','constructs diff (EXAMPLE)'], + var_names=['construct_mu','construct_sigma','constructs diff (EXAMPLE)'], point_estimate='mode', credible_interval=0.95, - ref_val=0) \ No newline at end of file + ref_val=0)