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

add seq indexing to PaddedBatch #132

Open Ilykuleshov opened 9 months ago

Ilykuleshov commented 9 months ago

Implements #131

ivkireev86 commented 9 months ago

Hello, @Ilykuleshov Thank you for the contribution. Could you please add unit test for your code It should be in ptls_tests/test_data_load/test_padded_batch.py file.