def iterator(self, data_type="train"):
raw_data = self.get_data_from_type(data_type)
return itertools.cycle(([self.onehot(data), data] for data in raw_data if data != []))
微信
公众号
扫码关注公众号