Closed Ilykuleshov closed 10 months ago
The property TrxEncoder.category_max_size breaks, since TrxEncoder.embeddings.values() are subclasses of nn.Embedding, and don't support dict-like access, required here
TrxEncoder.category_max_size
TrxEncoder.embeddings.values()
Also #129 with unit test for this bug
@Ilykuleshov , thank you for contribution!
The property
TrxEncoder.category_max_size
breaks, sinceTrxEncoder.embeddings.values()
are subclasses of nn.Embedding, and don't support dict-like access, required here