19 Commits

Author	SHA1	Message	Date
D-X-Y	f809f0a063	Remove un-used imports	2021-03-28 10:50:25 +00:00
D-X-Y	0386df7b16	Collect all contrib models in __init__ and add unit tests for init	2021-03-28 10:39:28 +00:00
D-X-Y	07434da8b0	Remove unused imports	2021-03-15 03:35:34 +00:00
D-X-Y	53a6b72ce5	Fix black errors	2021-03-15 03:09:52 +00:00
D-X-Y	a51dafcb4c	Remove unnecessary codes	2021-03-15 03:07:25 +00:00
D-X-Y	d5f9395e51	Fix Ghost BN bugs in TabNet and simplify its implementation	2021-03-14 07:25:09 +00:00
D-X-Y	1d435248e2	Add return for use_gpu..	2021-03-11 19:28:00 -08:00
D-X-Y	d38b8d6001	Fix bugs in use_gpu	2021-03-11 19:10:32 -08:00
D-X-Y	db59713d36	Add torch.no_grad for evaluation	2021-03-12 02:46:04 +00:00
D-X-Y	67fbdafe76	Fix many bugs in TabNet and use_gpu	2021-03-12 02:42:25 +00:00
D-X-Y	b99de068f8	Move save_path to get_or_create_path, and fix bugs in sfm / tabnet	2021-03-11 12:52:26 +00:00
D-X-Y	ca48345b29	Simplify count_parameters	2021-03-08 08:16:17 +00:00
D-X-Y	7bed3b4c2e	Fix python format by black	2021-03-08 06:39:03 +00:00
D-X-Y	73b7107ee8	Remove useless verbose kwarg	2021-03-07 00:52:27 -08:00
D-X-Y	49697b1f15	Show model size for pytorch models	2021-03-05 12:46:41 +00:00
Jactus	b84686b215	Update models to enable save/load	2021-02-05 13:14:12 +08:00
Alex Wang	2b6d16feb1	fix naming	2021-01-22 19:16:57 +08:00
Alex Wang	0a86a6f392	update format	2021-01-22 19:16:57 +08:00
Alex Wang	5da5ad4b9f	tabnet	2021-01-22 19:16:57 +08:00