Skip to content

Releases: dreamquark-ai/tabnet

v4.1.0

23 Jul 13:34
Compare
Choose a tag to compare

4.1.0 (2023-07-23)

Bug Fixes

  • 424 allow any np.intX as training target (63a8dba)
  • compute unsupervised loss using numpy (49bd61b)
  • custom loss using inplace operations (423f7c4)
  • disable ansi (60ec6bf)
  • feature importance not dependent from dataloader (5b19091)
  • README patience to 10 (fd2c73a)
  • replace std 0 by the mean or 1 if mean is 0 (ddf02da)
  • try to disable parallel install (c4963ad)
  • typo in pandas error (5ac5583)
  • update gpg key in docker file gpu (709fcb1)
  • upgrade the ressource size (fc59ea6)
  • use numpy std with bessel correction and test (3adaf4c)

Features

  • add augmentations inside the fit method (6d0485f)
  • add warm_start matching scikit-learn (d725101)
  • added conda install option (ca14b76), closes #346
  • disable tests in docker file gpu to save CI time (233f74e)
  • enable feature grouping for attention mechanism (bcae5f4)
  • enable torch 2.0 by relaxing poetry (bbd7a4e)
  • pretraining matches paper (5adb804)
  • raise error in case cat_dims and cat_idxs are incoherent (8c3b795)
  • update python (dea62b4)

v4.0

14 Sep 16:48
Compare
Choose a tag to compare

What's Changed

New Contributors

Full Changelog: v3.1.1...v4.0

v3.1.1

02 Feb 08:05
Compare
Choose a tag to compare

Bug Fixes

  • add preds_mapper to pretraining (76f2c85)

v3.1.0

12 Jan 18:17
Compare
Choose a tag to compare

Bug Fixes

Features

  • save and load preds_mapper (cab643b)

v3.0.0

15 Dec 12:44
Compare
Choose a tag to compare

Bug Fixes

  • checknan allow string as targets (855befc)
  • deactivate pin memory when device is cpu (bd0b96f)
  • fixed docstring issues (d216fbf)
  • load from cpu when saved on gpu (451bd86)

Features

  • add new default metrics (0fe5b72)
  • enable self supervised pretraining (d4af838)
  • mask-dependent loss (64052b0)

v2.0.1

15 Oct 21:00
Compare
Choose a tag to compare

Bug Fixes

  • add check for evalset dim (ba09980)
  • pin memory available for training only (28346c2)
  • specify device (46a301f)
  • torch.load map_location in Py36 fallback (63cb8c4)

v2.0.0

13 Oct 12:10
Compare
Choose a tag to compare

Bug Fixes

  • 1000 lines only when env=CI (c557349)
  • add map_location to torch load (c2b560e)
  • load_model fallback to BytesIO for Py3.6 (55c09e5)

Features

  • add check nan and inf (d871406)
  • add easy schedulers (0ae114f)
  • adding callbacks and metrics (1e0daec)
  • refacto models with metrics and callbacks (cc57d62)
  • speedups (5a01359)
  • TabNetMultiTaskClassifier (5764a43)
  • update readme and notebooks (9cb38d2)

v1.2.0

01 Jul 16:27
Compare
Choose a tag to compare

Bug Fixes

  • verbosity with schedulers (d6fbf90)

Features

  • add entmax as parameter (96c8a74)
  • allow other optimizer parameters (16d92d5)
  • allow weights sample for regression (d40b02f)
  • save and load tabnet models (9d2d8ae)
  • save params and easy loading (6e22393)

v1.1.0

02 Jun 15:46
Compare
Choose a tag to compare

Bug Fixes

  • allow zero layer (e3b5a04)
  • sort by cat_idx into embedding generator (9ab3ad5)
  • update forest_example notebook (8092324)

Features

  • add multi output regression (ffd7c28)
  • add num_workers and drop_last to fit parameters (313d074)
  • remove mask computations from forward (44d1a47)
  • speed boost and code simplification for GBN (1642909)

v1.0.6

20 Apr 08:13
Compare
Choose a tag to compare
chore: release v1.0.6