Method map_node

DIEN/data_utils.py:99–100

(self, x)

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

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

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