26 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	a51dafcb4c	Remove unnecessary codes	2021-03-15 03:07:25 +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	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	73b7107ee8	Remove useless verbose kwarg	2021-03-07 00:52:27 -08:00
D-X-Y	131f0e2e67	Add count_parameters for pytorch models in contrib	2021-03-05 12:07:43 +00:00
Jactus	b84686b215	Update models to enable save/load	2021-02-05 13:14:12 +08:00
Wendi Li	ebc31b9bdb	Update pytorch_lstm.py	2021-02-02 14:46:49 +08:00
Jactus	745b93138d	Update devices	2020-12-02 13:28:32 +08:00
Jactus	7f385345bb	Fix GPU	2020-12-02 13:25:29 +08:00
Jactus	d109d3d44e	Fix GPU	2020-12-02 12:23:42 +08:00
Jactus	77884db3a5	Fix and update run_all_model	2020-12-01 11:54:26 +08:00
lwwang1995	e0dad0e85a	Add random seed.	2020-12-01 09:31:22 +08:00
lwwang1995	1353e81b5b	Fix code with block.	2020-11-27 22:44:28 +08:00
lwwang1995	bebce24a7c	Update all baseline models.	2020-11-27 22:30:05 +08:00
Dong Zhou	55acac9fd5	black format	2020-11-27 10:42:20 +08:00
lwwang1995	6fb19eb58d	Update training setting.	2020-11-27 10:09:17 +08:00
Hong Zhang	5796363ecf	revise settings	2020-11-27 00:19:23 +08:00
Jactus	2a17062421	Update run_all_model and format	2020-11-26 13:49:51 +08:00
lwwang1995	5102566aad	Update GRU and LSTM model.	2020-11-26 13:35:07 +08:00
Jactus	a8ad2120c9	Update recordTemp and report	2020-11-19 16:50:16 +08:00
lwwang1995	e6a902c659	Add LSTM and Gats	2020-11-18 21:53:28 +08:00