def prepare_for_estimate(self, agents_for_estimation_storage, agents_for_estimation_table, agent_set, **kwargs): estimation_set = Dataset(in_storage = agents_for_estimation_storage, in_table_name=agents_for_estimation_table, id_name=agent_set.get_id_name(), dataset_name=agent_set.get_dataset_name()) hhs_est = HouseholdDataset(in_storage=agents_for_estimation_storage, in_table_name='households_for_estimation') self.dataset_pool.replace_dataset('household', hhs_est) self.dataset_pool.replace_dataset(estimation_set.get_dataset_name(), estimation_set) spec, index = ChoiceModel.prepare_for_estimate(self, estimation_set, **kwargs) return (spec, index, estimation_set)
def prepare_for_estimate_hh(self, estimation_storage, agents_for_estimation_table, agent_set, persons_for_estimation_table=None, **kwargs): estimation_set = Dataset(in_storage = estimation_storage, in_table_name=agents_for_estimation_table, id_name=agent_set.get_id_name(), dataset_name=agent_set.get_dataset_name()) if persons_for_estimation_table is not None: pers = PersonDataset(in_storage=estimation_storage, in_table_name=persons_for_estimation_table) self.dataset_pool.replace_dataset('person', pers) self.dataset_pool.replace_dataset(estimation_set.get_dataset_name(), estimation_set) spec, index = ChoiceModel.prepare_for_estimate(self, estimation_set, **kwargs) return (spec, index, estimation_set)