Published June 3, 2021 | Version 1.0.0
Software Open

PROCAT Set-to-Sequence Model

  • 1. IT University of Copenhagen

Description

This file contains a set-to-sequence model trained on the PROCAT dataset of product catalogues. It consists of a Set Transformer module that encodes the set of product offers, and a permutation learning module in the form of a Pointer Network that then implicitly clusters them into complementary sections and outputs the target catalogue structure.

References to the authors of the individual used models can be found in the paper describing the dataset, which is under review for the NeurIPS 2021 Dataset track as of June 2021, referenced in the repository with code for repeated experiments, which can be found here:

https://github.com/mateuszjurewicz/procat

The model requires using the appropriate version of PyTorch and importing the proper model definition, as shown in the provided jupyter notebooks.

Files

Files (39.5 MB)

Name Size Download all
md5:5866567767a0cd23a319c8410cfa25b5
39.5 MB Download

Additional details

Related works

Is supplemented by
Dataset: 10.6084/m9.figshare.14709507 (DOI)