Skip to content

Releases: dreamquark-ai/tabnet

v1.0.5

13 Mar 15:18
Compare
Choose a tag to compare

Bug Fixes

  • remove dead code for plots (f96795f)

Features

  • switch to sparse matrix trick (98910bc)

v1.0.4

28 Feb 16:36
Compare
Choose a tag to compare

Bug Fixes

  • allow smaller different nshared and nindependent (4b365a7)
  • sparsemax on train and predict epoch (6f7c0e0)

v1.0.3

07 Feb 16:54
Compare
Choose a tag to compare

Bug Fixes

  • map class predictions for XGB results (3747e2f)

Features

  • fix shared layers with independent batchnorm (5f0e43f)

v1.0.2

03 Feb 16:22
Compare
Choose a tag to compare

Bug Fixes

  • multiclass prediction mapper (2317c5c)
  • remove deepcopy from shared blocks (123932a)

v1.0.1

20 Jan 15:28
Compare
Choose a tag to compare

Bug Fixes

  • regression: fix scheduler (01e46b7)
  • fixing Dockerfile for poetry 1.0 (6c5cdec)
  • importance indexing fixed (a8382c3)
  • local explain all batches (91461fb)
  • regression gpu integration an typos (269b4c5)
  • resolve timer issue and warnings (ecd2cd9)

Features

v1.0.0

03 Dec 11:40
Compare
Choose a tag to compare

Bug Fixes

  • deps: update dependency numpy to v1.17.3 (eff6555)
  • deps: update dependency numpy to v1.17.4 (a80cf29)
  • deps: update dependency torch to v1.3.1 (18ec79b)
  • deps: update dependency tqdm to v4.37.0 (f8f04e7)
  • deps: update dependency tqdm to v4.38.0 (0bf45d2)
  • functional balanced version (fab7f16)
  • remove torch warnings (index should be bool) (f5817cf)

Features

  • add gpu dockerfile and adapt makefile (8d14406)
  • update notebooks for new model format (43e2693)

v0.1.2

06 Nov 08:48
Compare
Choose a tag to compare

Bug Fixes

  • add softmax to predict_proba (bea966f)
  • correct code linting (ae3098c)
  • float type when output_dim=1 (7bb7dfd)

Features