def load_data_batch(self, attrs_batch, stages_batch):
"""load data into model"""
set_shared(self.attributes, attrs_batch)
set_shared(self.disease_stages, stages_batch)
def set_data_batch(self,attrs_batch,categories_batch):
"""load data into model"""
set_shared(self.attributes,attrs_batch)
set_shared(self.categories,categories_batch)