Skip to content

Releases: xiahouzuoxin/torchctr

Move to polars, improve Trainer

06 Feb 07:52

Choose a tag to compare

Full Changelog: v0.1.3...v0.2.0

  • FeatureTransform:
    • polars is preferred over pandas
    • add feature cross, feature pre-process by polars' Expr
    • fix lots of bugs
  • Trainer:
    • add accelerate to support the trainer on GPU
    • add tensorboard support
    • save only the model state dict instead of the pickle file to help loading the model on another device

v0.1.3

15 Nov 07:46

Choose a tag to compare

The last version that feature transform by pandas.

Full Changelog: v0.1.2...v0.1.3

v0.1.2

14 Aug 02:45

Choose a tag to compare

  • Fixed the OOM problem of DataFrameDataset by replacing list with numpy, and process all data to numpy before __get_item__
  • Default value changes of DataFrameDataset:
    • is_raw parameter default to True
    • list_padding_value parameter default to None. Therefore, you need to set the list_padding_value when init DataFrameDataset or using collate_fn in DataFrameDataset when data loading if you have var-length sequence features
  • Support string input type of sequence features (comma split)

Full Changelog: v0.1.1...v0.1.2

v0.1.1

22 Jul 07:48

Choose a tag to compare

Full Changelog: v0.1.0...v0.1.1

Fixed a minor problem when input sequence features are the same length.

v0.1.0

18 Jul 06:54

Choose a tag to compare

Full Changelog: https://github.com/xiahouzuoxin/torchrec/commits/v0.1.0

The first release with all features implemented as README.md.