def __init__(self, data, **kwargs):
super(EstimationSingleOptimization, self).__init__(data, **kwargs)
#create an HDDM model for each subject
grouped_data = data.groupby('subj_idx')
self.models = []
for subj_idx, subj_data in grouped_data:
model = my_hddm.HDDMGamma(subj_data.to_records(),
is_group_model=False,
**kwargs)
self.models.append(model)
def init_model(self, data):
self.model = my_hddm.HDDMGamma(data,
group_only_nodes=['sz', 'st', 'sv'],
**self.init_kwargs)
def __init__(self, data, **kwargs):
super(self.__class__, self).__init__(data, **kwargs)
self.model = my_hddm.HDDMGamma(data, **kwargs)