search

dllllb / pytorch-lifestream

A library built upon PyTorch for building embeddings on discrete event sequences using self-supervision
Apache License 2.0
215 stars 46 forks source link

TrxEncoder bug #124

Closed Ilykuleshov closed 10 months ago

Ilykuleshov commented 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

ivkireev86 commented 10 months ago

Also #129 with unit test for this bug

ivkireev86 commented 10 months ago

@Ilykuleshov , thank you for contribution!