Add Tabnet support

.github/workflows/test_full.yml

@@ -27,8 +27,8 @@ jobs:
         if: ${{ matrix.os == 'macos-latest' }}
       - name: Install dependencies
         run: |
+            pip install pip==23.0.1
             pip install -r prereq.txt
-            pip install --upgrade pip
       - name: Test Core
         run: |
           pip install .[testing]

.github/workflows/test_pr.yml

@@ -54,8 +54,8 @@ jobs:
         if: ${{ matrix.os == 'macos-latest' }}
       - name: Install dependencies
         run: |
+            pip install pip==23.0.1
             pip install -r prereq.txt
-            pip install --upgrade pip
       - name: Test Core
         run: |
           pip install .[testing]

.github/workflows/test_tutorials.yml

@@ -32,8 +32,8 @@ jobs:
         if: ${{ matrix.os == 'macos-latest' }}
       - name: Install dependencies
         run: |
+            pip install pip==23.0.1
             pip install -r prereq.txt
-            pip install --upgrade pip
 
             pip install .[all]
 

src/synthcity/plugins/core/models/factory.py

@@ -20,9 +20,9 @@
     DatetimeEncoder,
     FeatureEncoder,
     GaussianQuantileTransformer,
-    LabelEncoder,
     MinMaxScaler,
     OneHotEncoder,
+    OrdinalEncoder,
     RobustScaler,
     StandardScaler,
 )
@@ -74,7 +74,7 @@
 FEATURE_ENCODERS = dict(
     datetime=DatetimeEncoder,
     onehot=OneHotEncoder,
-    label=LabelEncoder,
+    ordinal=OrdinalEncoder,
     standard=StandardScaler,
     minmax=MinMaxScaler,
     robust=RobustScaler,

src/synthcity/plugins/core/models/tabnet.py

@@ -1,3 +1,8 @@
+# TabNet: Attentive Interpretable Tabular Learning
+# Reference:
+# - https://arxiv.org/pdf/1908.07442.pdf
+# - https://github.com/dreamquark-ai/tabnet
+
 # stdlib
 from typing import List, Optional, Tuple