I wonder whether the cluster number K is manual setup or auto-calculated?

OctoberChang / X-Transformer

X-Transformer: Taming Pretrained Transformers for eXtreme Multi-label Text Classification

BSD 3-Clause "New" or "Revised" License

134 stars 28 forks source link

I wonder whether the cluster number K is manual setup or auto-calculated? #15

Closed Gpwner closed 3 years ago

Gpwner commented 3 years ago

Any help will be appreciate！

OctoberChang commented 3 years ago

The number of cluster K is manually set. The heuristic is we control how the maximum number of labels within a cluster. See more details in Sec 3.1 of PECOS paper (https://arxiv.org/pdf/2010.05878.pdf).