Add TabNet config

2026-07-11 06:46:56 +08:00 · 2020-11-25 11:16:01 +08:00
parent 0e2c2fcd7f
commit 991c6195bd
9 changed files with 114 additions and 34 deletions
--- a/setup.py
+++ b/setup.py
@@ -58,6 +58,7 @@ REQUIRED = [
    "joblib>=0.17.0",
    "fire>=0.3.1",
    "ruamel.yaml>=0.16.12",
+    "pytorch-tabnet>=2.0.1",
 ]

 # Numpy include