Method map_cate

DIEN/data_iterator.py:90–91

(self, x)

        return int(self.user_dict.get(x, -1))

    def map_cate(self, x):
        return int(self.cate_dict.get(x, -1))

    def map_shop(self, x):
        return int(self.shop_dict.get(x, -1))