Method map_node

DIEN/data_iterator.py:96–97

(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))