Method map_product

DIEN/data_iterator.py:99–100

(self, x)

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

    def map_product(self, x):
        return int(self.product_dict.get(x, -1))

    def map_brand(self, x):
        return int(self.brand_dict.get(x, -1))