def prepare_batch_data(self, batch_generator, data_inst): batch_generator.initialize_batch_generator(data_inst) batch_data_generator = batch_generator.generate_batch_data() for batch_data in batch_data_generator: batch_x = self._load_data(batch_data) self.data_x.append(batch_x) self.set_partition(data_inst)
def prepare_batch_data(self, batch_generator, data_inst): batch_generator.initialize_batch_generator(data_inst, self.batch_size) batch_data_generator = batch_generator.generate_batch_data() for batch_data in batch_data_generator: keys, batch_x, batch_y = self._load_data(batch_data) self.data_x.append(batch_x) self.data_y.append(batch_y) self.data_keys.append(keys) self._convert_label() self.set_partition(data_inst)