Alcoholrithm / TabularS3L

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

Merge dev to main #14

Closed Alcoholrithm closed 5 months ago

Alcoholrithm commented 5 months ago
  1. Optimize SubTab, SCARF, and SwitchTab
  2. Fix second_phase_loss of VIME functional
  3. Add an option to freeze the encoder in the set_second_phase method of Lightning modules
  4. Replace np.float_ with np.float64 in data_utils