search

Alcoholrithm / TabularS3L

A PyTorch Lightning-based library for self- and semi-supervised learning on tabular data.
MIT License
27 stars 2 forks source link

Roadmap for v0.50 #11

Closed Alcoholrithm closed 5 months ago

Alcoholrithm commented 5 months ago

I'm planning to undertake a significant refactoring of this repository to optimize the codebase and improve the maintainability and scalability of this project. This refactoring will include moving forward and loss calculation logic from the current Lightning modules to the functional module.

  1. Optimize the codebase

    • [x] utils
    • [x] DAE
    • [x] VIME
    • [x] SubTab
    • [x] SCARF
    • [x] SwitchTab

  2. Move the forward and loss calculation logic from our current Lightning modules to the functional module.

    • [x] DAE
    • [x] VIME
    • [x] SubTab
    • [x] SCARF
    • [x] SwitchTab

  3. Additional features

    • [x] Add a freeze encoder option to the set_second_phase method of Lightning modules.