diff --git a/wids/stg_wids_ctgan.model/args.json b/wids/stg_wids_ctgan.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..0ecfa3ce9625c599e90aff07963f738f98b67d2a
--- /dev/null
+++ b/wids/stg_wids_ctgan.model/args.json
@@ -0,0 +1,35 @@
+{
+    "name": "stg",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.09910064387516969,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "lam": 0.03675976401215312,
+    "hidden_dims": [
+        500,
+        400,
+        20
+    ],
+    "num_classes": 2,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "stg",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/stg_wids_ctgan.model/scaler.json b/wids/stg_wids_ctgan.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/stg_wids_ctgan.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/stg_wids_ctgan.model/weights.pt b/wids/stg_wids_ctgan.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..504349ae76bde5a510a0a36968ef98dc4711f482
--- /dev/null
+++ b/wids/stg_wids_ctgan.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6aeaea044138ad14fa344c8ca19180a8f9f63c806aa227537bac382e4967a65a
+size 1388231
diff --git a/wids/stg_wids_ctgan_madry.model/args.json b/wids/stg_wids_ctgan_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..0ecfa3ce9625c599e90aff07963f738f98b67d2a
--- /dev/null
+++ b/wids/stg_wids_ctgan_madry.model/args.json
@@ -0,0 +1,35 @@
+{
+    "name": "stg",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.09910064387516969,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "lam": 0.03675976401215312,
+    "hidden_dims": [
+        500,
+        400,
+        20
+    ],
+    "num_classes": 2,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "stg",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/stg_wids_ctgan_madry.model/scaler.json b/wids/stg_wids_ctgan_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/stg_wids_ctgan_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/stg_wids_ctgan_madry.model/weights.pt b/wids/stg_wids_ctgan_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..f673ad87b1479ec019e74f92160d74bd75616da5
--- /dev/null
+++ b/wids/stg_wids_ctgan_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:aa1bda7fec7f16e9dc6e59d1af886af8cccb9ca21c3977a6f2c6947cb609e198
+size 1388231
diff --git a/wids/stg_wids_cutmix.model/args.json b/wids/stg_wids_cutmix.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..0ecfa3ce9625c599e90aff07963f738f98b67d2a
--- /dev/null
+++ b/wids/stg_wids_cutmix.model/args.json
@@ -0,0 +1,35 @@
+{
+    "name": "stg",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.09910064387516969,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "lam": 0.03675976401215312,
+    "hidden_dims": [
+        500,
+        400,
+        20
+    ],
+    "num_classes": 2,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "stg",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/stg_wids_cutmix.model/scaler.json b/wids/stg_wids_cutmix.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/stg_wids_cutmix.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/stg_wids_cutmix.model/weights.pt b/wids/stg_wids_cutmix.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..effbca5526fb3cf90eaac5d32412bca1d87ce2cc
--- /dev/null
+++ b/wids/stg_wids_cutmix.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3b3b83a2cb43282ced50017113e812d224a20554e6d9b3d77eb56fe7d735cb02
+size 1388231
diff --git a/wids/stg_wids_cutmix_madry.model/args.json b/wids/stg_wids_cutmix_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..0ecfa3ce9625c599e90aff07963f738f98b67d2a
--- /dev/null
+++ b/wids/stg_wids_cutmix_madry.model/args.json
@@ -0,0 +1,35 @@
+{
+    "name": "stg",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.09910064387516969,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "lam": 0.03675976401215312,
+    "hidden_dims": [
+        500,
+        400,
+        20
+    ],
+    "num_classes": 2,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "stg",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/stg_wids_cutmix_madry.model/scaler.json b/wids/stg_wids_cutmix_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/stg_wids_cutmix_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/stg_wids_cutmix_madry.model/weights.pt b/wids/stg_wids_cutmix_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..b496c357763a0fd9d9641966a4736a2ea6982553
--- /dev/null
+++ b/wids/stg_wids_cutmix_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4297d8f6fefdf7ed6f38d86d447494ab9323efe90b3617e1a54090a289181f87
+size 1388231
diff --git a/wids/stg_wids_default.model/args.json b/wids/stg_wids_default.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..c985d52242339cf892ed47d9442b28e0b5114b4a
--- /dev/null
+++ b/wids/stg_wids_default.model/args.json
@@ -0,0 +1,32 @@
+{
+    "name": "stg",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.09910064387516969,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "lam": 0.03675976401215312,
+    "hidden_dims": [
+        500,
+        400,
+        20
+    ],
+    "num_classes": 2,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "stg",
+    "dataset": "wids",
+    "num_splits": 5,
+    "seed": 42,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/stg_wids_default.model/weights.pt b/wids/stg_wids_default.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..f959a30ff7ad1abe68b33213596d6c3a7f139dab
--- /dev/null
+++ b/wids/stg_wids_default.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0480d1e654f94f93320feedae8edd9272125d2b9ef942e068afc13cf6b18b16a
+size 1388167
diff --git a/wids/stg_wids_dist.model/args.json b/wids/stg_wids_dist.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..c985d52242339cf892ed47d9442b28e0b5114b4a
--- /dev/null
+++ b/wids/stg_wids_dist.model/args.json
@@ -0,0 +1,32 @@
+{
+    "name": "stg",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.09910064387516969,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "lam": 0.03675976401215312,
+    "hidden_dims": [
+        500,
+        400,
+        20
+    ],
+    "num_classes": 2,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "stg",
+    "dataset": "wids",
+    "num_splits": 5,
+    "seed": 42,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/stg_wids_dist.model/weights.pt b/wids/stg_wids_dist.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..909bf9f892a3cf061fac47da473e8aa4f193cd3f
--- /dev/null
+++ b/wids/stg_wids_dist.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:284f877f78ff3313eb681ceb88b1dde85920959a31fdf46394c6fb66e0f0106f
+size 1388167
diff --git a/wids/stg_wids_goggle.model/args.json b/wids/stg_wids_goggle.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..0ecfa3ce9625c599e90aff07963f738f98b67d2a
--- /dev/null
+++ b/wids/stg_wids_goggle.model/args.json
@@ -0,0 +1,35 @@
+{
+    "name": "stg",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.09910064387516969,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "lam": 0.03675976401215312,
+    "hidden_dims": [
+        500,
+        400,
+        20
+    ],
+    "num_classes": 2,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "stg",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/stg_wids_goggle.model/scaler.json b/wids/stg_wids_goggle.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/stg_wids_goggle.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/stg_wids_goggle.model/weights.pt b/wids/stg_wids_goggle.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..5919c3de71924ed69583616a66f3f473fedae86c
--- /dev/null
+++ b/wids/stg_wids_goggle.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:601c37ffd8799bb8903d3b9a681c8c63561b538422487efeca06b4f59ed97055
+size 1388231
diff --git a/wids/stg_wids_goggle_madry.model/args.json b/wids/stg_wids_goggle_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..0ecfa3ce9625c599e90aff07963f738f98b67d2a
--- /dev/null
+++ b/wids/stg_wids_goggle_madry.model/args.json
@@ -0,0 +1,35 @@
+{
+    "name": "stg",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.09910064387516969,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "lam": 0.03675976401215312,
+    "hidden_dims": [
+        500,
+        400,
+        20
+    ],
+    "num_classes": 2,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "stg",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/stg_wids_goggle_madry.model/scaler.json b/wids/stg_wids_goggle_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/stg_wids_goggle_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/stg_wids_goggle_madry.model/weights.pt b/wids/stg_wids_goggle_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..5e6835b64fd569ce2d9c8a6ad328f905f023b561
--- /dev/null
+++ b/wids/stg_wids_goggle_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:08e3d07a5ce37d482798d30bbf4c97c26155213fa05313ab06cc3d5ed788db4a
+size 1388231
diff --git a/wids/stg_wids_madry.model/args.json b/wids/stg_wids_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..c985d52242339cf892ed47d9442b28e0b5114b4a
--- /dev/null
+++ b/wids/stg_wids_madry.model/args.json
@@ -0,0 +1,32 @@
+{
+    "name": "stg",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.09910064387516969,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "lam": 0.03675976401215312,
+    "hidden_dims": [
+        500,
+        400,
+        20
+    ],
+    "num_classes": 2,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "stg",
+    "dataset": "wids",
+    "num_splits": 5,
+    "seed": 42,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/stg_wids_madry.model/weights.pt b/wids/stg_wids_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..8515b920ac6b31e5ac7b7e7ceeee24e03156ec77
--- /dev/null
+++ b/wids/stg_wids_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9824645d694e00dd807de8ec8ac0f7fb274c66a110d799faf97943b4ac0628a6
+size 1388167
diff --git a/wids/stg_wids_subset.model/args.json b/wids/stg_wids_subset.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..c985d52242339cf892ed47d9442b28e0b5114b4a
--- /dev/null
+++ b/wids/stg_wids_subset.model/args.json
@@ -0,0 +1,32 @@
+{
+    "name": "stg",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.09910064387516969,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "lam": 0.03675976401215312,
+    "hidden_dims": [
+        500,
+        400,
+        20
+    ],
+    "num_classes": 2,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "stg",
+    "dataset": "wids",
+    "num_splits": 5,
+    "seed": 42,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/stg_wids_subset.model/weights.pt b/wids/stg_wids_subset.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..a6cdeca4ea5bc1219a0ea93ed8f8256b5ba81252
--- /dev/null
+++ b/wids/stg_wids_subset.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e2d5641640028489f39ae2bcd395c073e2e496b3b8ed3ede9789e3a1db65ad8a
+size 1388167
diff --git a/wids/stg_wids_tablegan.model/args.json b/wids/stg_wids_tablegan.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..0ecfa3ce9625c599e90aff07963f738f98b67d2a
--- /dev/null
+++ b/wids/stg_wids_tablegan.model/args.json
@@ -0,0 +1,35 @@
+{
+    "name": "stg",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.09910064387516969,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "lam": 0.03675976401215312,
+    "hidden_dims": [
+        500,
+        400,
+        20
+    ],
+    "num_classes": 2,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "stg",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/stg_wids_tablegan.model/scaler.json b/wids/stg_wids_tablegan.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/stg_wids_tablegan.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/stg_wids_tablegan.model/weights.pt b/wids/stg_wids_tablegan.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..e4c3f8b8ce19db194a9e5fb27a57f889cc78739e
--- /dev/null
+++ b/wids/stg_wids_tablegan.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8cdb16cd93e040fd81b31e46e76e3f3ad78519c170f4b5d102c82bd7fe44a642
+size 1388231
diff --git a/wids/stg_wids_tablegan_madry.model/args.json b/wids/stg_wids_tablegan_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..0ecfa3ce9625c599e90aff07963f738f98b67d2a
--- /dev/null
+++ b/wids/stg_wids_tablegan_madry.model/args.json
@@ -0,0 +1,35 @@
+{
+    "name": "stg",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.09910064387516969,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "lam": 0.03675976401215312,
+    "hidden_dims": [
+        500,
+        400,
+        20
+    ],
+    "num_classes": 2,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "stg",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/stg_wids_tablegan_madry.model/scaler.json b/wids/stg_wids_tablegan_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/stg_wids_tablegan_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/stg_wids_tablegan_madry.model/weights.pt b/wids/stg_wids_tablegan_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..f18b763f825657fa3ad4e8b53933496a71ca01bc
--- /dev/null
+++ b/wids/stg_wids_tablegan_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2f8cc1c694d6d9c926f68c04c128f9737dff1f20b57c00cf1ce43cd46059b263
+size 1388231
diff --git a/wids/stg_wids_tvae.model/args.json b/wids/stg_wids_tvae.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..0ecfa3ce9625c599e90aff07963f738f98b67d2a
--- /dev/null
+++ b/wids/stg_wids_tvae.model/args.json
@@ -0,0 +1,35 @@
+{
+    "name": "stg",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.09910064387516969,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "lam": 0.03675976401215312,
+    "hidden_dims": [
+        500,
+        400,
+        20
+    ],
+    "num_classes": 2,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "stg",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/stg_wids_tvae.model/scaler.json b/wids/stg_wids_tvae.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/stg_wids_tvae.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/stg_wids_tvae.model/weights.pt b/wids/stg_wids_tvae.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..031bedec8a387304c92ae33ed89da81d18d15e56
--- /dev/null
+++ b/wids/stg_wids_tvae.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d6b3ac5d03bf62082d4d9195306fe247cd6d500fd5ccc007bc0d3adb11090d64
+size 1388231
diff --git a/wids/stg_wids_tvae_madry.model/args.json b/wids/stg_wids_tvae_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..0ecfa3ce9625c599e90aff07963f738f98b67d2a
--- /dev/null
+++ b/wids/stg_wids_tvae_madry.model/args.json
@@ -0,0 +1,35 @@
+{
+    "name": "stg",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.09910064387516969,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "lam": 0.03675976401215312,
+    "hidden_dims": [
+        500,
+        400,
+        20
+    ],
+    "num_classes": 2,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "stg",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/stg_wids_tvae_madry.model/scaler.json b/wids/stg_wids_tvae_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/stg_wids_tvae_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/stg_wids_tvae_madry.model/weights.pt b/wids/stg_wids_tvae_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..b44c4c95cc2002c4d1b30dc5b4f8777f31908575
--- /dev/null
+++ b/wids/stg_wids_tvae_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a5f458c72dc14d5a67d64a219f425b27c145c3ba982f7c540b013f79d6b1db4a
+size 1388231
diff --git a/wids/stg_wids_wgan.model/args.json b/wids/stg_wids_wgan.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..0ecfa3ce9625c599e90aff07963f738f98b67d2a
--- /dev/null
+++ b/wids/stg_wids_wgan.model/args.json
@@ -0,0 +1,35 @@
+{
+    "name": "stg",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.09910064387516969,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "lam": 0.03675976401215312,
+    "hidden_dims": [
+        500,
+        400,
+        20
+    ],
+    "num_classes": 2,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "stg",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/stg_wids_wgan.model/scaler.json b/wids/stg_wids_wgan.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/stg_wids_wgan.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/stg_wids_wgan.model/weights.pt b/wids/stg_wids_wgan.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..8017348afa299969b41c6ad7c29acde626ea244d
--- /dev/null
+++ b/wids/stg_wids_wgan.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f79a7c7af536b2fc56f4d42b2c05f2944af69344d2d52894b939c21a2abee85e
+size 1388231
diff --git a/wids/stg_wids_wgan_madry.model/args.json b/wids/stg_wids_wgan_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..0ecfa3ce9625c599e90aff07963f738f98b67d2a
--- /dev/null
+++ b/wids/stg_wids_wgan_madry.model/args.json
@@ -0,0 +1,35 @@
+{
+    "name": "stg",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.09910064387516969,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "lam": 0.03675976401215312,
+    "hidden_dims": [
+        500,
+        400,
+        20
+    ],
+    "num_classes": 2,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "stg",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/stg_wids_wgan_madry.model/scaler.json b/wids/stg_wids_wgan_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/stg_wids_wgan_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/stg_wids_wgan_madry.model/weights.pt b/wids/stg_wids_wgan_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..ea3e4656dae74d14a43d3b0f3801ef7c2e4daae3
--- /dev/null
+++ b/wids/stg_wids_wgan_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7ba905cc59b52a8faf876785c5d7698e913dffea9004891e15567ca321447955
+size 1388231
diff --git a/wids/tabnet_wids_ctgan.model/args.json b/wids/tabnet_wids_ctgan.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..8b34de182ea8063dc6d63880525d6c53a2e2d8be
--- /dev/null
+++ b/wids/tabnet_wids_ctgan.model/args.json
@@ -0,0 +1,37 @@
+{
+    "name": "tabnet",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_d": 15,
+    "n_steps": 4,
+    "gamma": 1.5764779589451128,
+    "cat_emb_dim": 3,
+    "n_independent": 3,
+    "n_shared": 3,
+    "momentum": 0.04967611500740168,
+    "mask_type": "entmax",
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "tabnet",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabnet_wids_ctgan.model/scaler.json b/wids/tabnet_wids_ctgan.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/tabnet_wids_ctgan.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/tabnet_wids_ctgan.model/weights.pt b/wids/tabnet_wids_ctgan.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..6001443cd30a6bcdde043a3d81f77543d4228331
--- /dev/null
+++ b/wids/tabnet_wids_ctgan.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:57c1b25f567e7d9a133bdc3a71440c18b37542a12e226f4a1a6e9b46fdf47bb8
+size 302425
diff --git a/wids/tabnet_wids_ctgan_madry.model/args.json b/wids/tabnet_wids_ctgan_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..8b34de182ea8063dc6d63880525d6c53a2e2d8be
--- /dev/null
+++ b/wids/tabnet_wids_ctgan_madry.model/args.json
@@ -0,0 +1,37 @@
+{
+    "name": "tabnet",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_d": 15,
+    "n_steps": 4,
+    "gamma": 1.5764779589451128,
+    "cat_emb_dim": 3,
+    "n_independent": 3,
+    "n_shared": 3,
+    "momentum": 0.04967611500740168,
+    "mask_type": "entmax",
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "tabnet",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabnet_wids_ctgan_madry.model/scaler.json b/wids/tabnet_wids_ctgan_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/tabnet_wids_ctgan_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/tabnet_wids_ctgan_madry.model/weights.pt b/wids/tabnet_wids_ctgan_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..39be3ea59553e03149e45b1243e280cf6e970784
--- /dev/null
+++ b/wids/tabnet_wids_ctgan_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:61053e58802c2ad14d258da9e6c32ba4a6de9c76e7cc1a506d03d363fe7921cb
+size 302425
diff --git a/wids/tabnet_wids_cutmix.model/args.json b/wids/tabnet_wids_cutmix.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..8b34de182ea8063dc6d63880525d6c53a2e2d8be
--- /dev/null
+++ b/wids/tabnet_wids_cutmix.model/args.json
@@ -0,0 +1,37 @@
+{
+    "name": "tabnet",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_d": 15,
+    "n_steps": 4,
+    "gamma": 1.5764779589451128,
+    "cat_emb_dim": 3,
+    "n_independent": 3,
+    "n_shared": 3,
+    "momentum": 0.04967611500740168,
+    "mask_type": "entmax",
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "tabnet",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabnet_wids_cutmix.model/scaler.json b/wids/tabnet_wids_cutmix.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/tabnet_wids_cutmix.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/tabnet_wids_cutmix.model/weights.pt b/wids/tabnet_wids_cutmix.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..eeab1262d84a71f48d09fb4a1a6d1ec987d0060f
--- /dev/null
+++ b/wids/tabnet_wids_cutmix.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:df78629921ab0ff3c0bae420eb191bbed29c547c1e4af1626101c900ac9bf1a4
+size 302425
diff --git a/wids/tabnet_wids_cutmix_madry.model/args.json b/wids/tabnet_wids_cutmix_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..8b34de182ea8063dc6d63880525d6c53a2e2d8be
--- /dev/null
+++ b/wids/tabnet_wids_cutmix_madry.model/args.json
@@ -0,0 +1,37 @@
+{
+    "name": "tabnet",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_d": 15,
+    "n_steps": 4,
+    "gamma": 1.5764779589451128,
+    "cat_emb_dim": 3,
+    "n_independent": 3,
+    "n_shared": 3,
+    "momentum": 0.04967611500740168,
+    "mask_type": "entmax",
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "tabnet",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabnet_wids_cutmix_madry.model/scaler.json b/wids/tabnet_wids_cutmix_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/tabnet_wids_cutmix_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/tabnet_wids_cutmix_madry.model/weights.pt b/wids/tabnet_wids_cutmix_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..30ed4dfbc8b5269196510440ad669e05b9e03d2a
--- /dev/null
+++ b/wids/tabnet_wids_cutmix_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8088281e0249df6284a0799229e686f7479bab9b2e451d65461b6262b94219f5
+size 302425
diff --git a/wids/tabnet_wids_default.model/args.json b/wids/tabnet_wids_default.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..29b612d40d0efcf757f934599db2936d533084fa
--- /dev/null
+++ b/wids/tabnet_wids_default.model/args.json
@@ -0,0 +1,34 @@
+{
+    "name": "tabnet",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "num_classes": 2,
+    "n_d": 15,
+    "n_steps": 4,
+    "gamma": 1.5764779589451128,
+    "cat_emb_dim": 3,
+    "n_independent": 3,
+    "n_shared": 3,
+    "momentum": 0.04967611500740168,
+    "mask_type": "entmax",
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "tabnet",
+    "dataset": "wids",
+    "num_splits": 5,
+    "seed": 42,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabnet_wids_default.model/weights.pt b/wids/tabnet_wids_default.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..eb9e41bff91bce3b7ac8eb78000b9d80fd2030ff
--- /dev/null
+++ b/wids/tabnet_wids_default.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cd1f04e1d4afbc7713369f3bdbc9dee10972490834b9fccfa51e3bad1a5036c7
+size 302425
diff --git a/wids/tabnet_wids_dist.model/args.json b/wids/tabnet_wids_dist.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..29b612d40d0efcf757f934599db2936d533084fa
--- /dev/null
+++ b/wids/tabnet_wids_dist.model/args.json
@@ -0,0 +1,34 @@
+{
+    "name": "tabnet",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "num_classes": 2,
+    "n_d": 15,
+    "n_steps": 4,
+    "gamma": 1.5764779589451128,
+    "cat_emb_dim": 3,
+    "n_independent": 3,
+    "n_shared": 3,
+    "momentum": 0.04967611500740168,
+    "mask_type": "entmax",
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "tabnet",
+    "dataset": "wids",
+    "num_splits": 5,
+    "seed": 42,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabnet_wids_dist.model/weights.pt b/wids/tabnet_wids_dist.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..7d6b5ec07e3683f53fa4ed8c5e009f1c4621c1fe
--- /dev/null
+++ b/wids/tabnet_wids_dist.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:792426846631c726702b032ab7ce7b7af098fb67a086175dacfb6fce516b1f10
+size 302425
diff --git a/wids/tabnet_wids_goggle.model/args.json b/wids/tabnet_wids_goggle.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..8b34de182ea8063dc6d63880525d6c53a2e2d8be
--- /dev/null
+++ b/wids/tabnet_wids_goggle.model/args.json
@@ -0,0 +1,37 @@
+{
+    "name": "tabnet",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_d": 15,
+    "n_steps": 4,
+    "gamma": 1.5764779589451128,
+    "cat_emb_dim": 3,
+    "n_independent": 3,
+    "n_shared": 3,
+    "momentum": 0.04967611500740168,
+    "mask_type": "entmax",
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "tabnet",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabnet_wids_goggle.model/scaler.json b/wids/tabnet_wids_goggle.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/tabnet_wids_goggle.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/tabnet_wids_goggle.model/weights.pt b/wids/tabnet_wids_goggle.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..cd61fc9a9a5f34c3859d44bbe199d34965c51c15
--- /dev/null
+++ b/wids/tabnet_wids_goggle.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b8665ac2813535e7c10a823a33329379e7f5d6748d8d2c0417db2fc5152389fe
+size 302425
diff --git a/wids/tabnet_wids_goggle_madry.model/args.json b/wids/tabnet_wids_goggle_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..8b34de182ea8063dc6d63880525d6c53a2e2d8be
--- /dev/null
+++ b/wids/tabnet_wids_goggle_madry.model/args.json
@@ -0,0 +1,37 @@
+{
+    "name": "tabnet",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_d": 15,
+    "n_steps": 4,
+    "gamma": 1.5764779589451128,
+    "cat_emb_dim": 3,
+    "n_independent": 3,
+    "n_shared": 3,
+    "momentum": 0.04967611500740168,
+    "mask_type": "entmax",
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "tabnet",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabnet_wids_goggle_madry.model/scaler.json b/wids/tabnet_wids_goggle_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/tabnet_wids_goggle_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/tabnet_wids_goggle_madry.model/weights.pt b/wids/tabnet_wids_goggle_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..847986c709a2d4bd382ff2e444f4f1cae44c3d84
--- /dev/null
+++ b/wids/tabnet_wids_goggle_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ccf0243c9c4b96378ea4c673518740a703acdeb7c999f2c8193e36e455f93f86
+size 302425
diff --git a/wids/tabnet_wids_madry.model/args.json b/wids/tabnet_wids_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..29b612d40d0efcf757f934599db2936d533084fa
--- /dev/null
+++ b/wids/tabnet_wids_madry.model/args.json
@@ -0,0 +1,34 @@
+{
+    "name": "tabnet",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "num_classes": 2,
+    "n_d": 15,
+    "n_steps": 4,
+    "gamma": 1.5764779589451128,
+    "cat_emb_dim": 3,
+    "n_independent": 3,
+    "n_shared": 3,
+    "momentum": 0.04967611500740168,
+    "mask_type": "entmax",
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "tabnet",
+    "dataset": "wids",
+    "num_splits": 5,
+    "seed": 42,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabnet_wids_madry.model/weights.pt b/wids/tabnet_wids_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..a9e777c0a01337de362f1fe4ecc13d164a93209d
--- /dev/null
+++ b/wids/tabnet_wids_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:34a70ebe3afa06b8ddbd55dad28177b8e24f11541776af3e465f6142e894614f
+size 302425
diff --git a/wids/tabnet_wids_subset.model/args.json b/wids/tabnet_wids_subset.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..29b612d40d0efcf757f934599db2936d533084fa
--- /dev/null
+++ b/wids/tabnet_wids_subset.model/args.json
@@ -0,0 +1,34 @@
+{
+    "name": "tabnet",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "num_classes": 2,
+    "n_d": 15,
+    "n_steps": 4,
+    "gamma": 1.5764779589451128,
+    "cat_emb_dim": 3,
+    "n_independent": 3,
+    "n_shared": 3,
+    "momentum": 0.04967611500740168,
+    "mask_type": "entmax",
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "tabnet",
+    "dataset": "wids",
+    "num_splits": 5,
+    "seed": 42,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabnet_wids_subset.model/weights.pt b/wids/tabnet_wids_subset.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..d88109d161ba3da5c76e2d21167365fc15f205ef
--- /dev/null
+++ b/wids/tabnet_wids_subset.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5403a0a367c3c0c235431513258fab9408ff22cfce0f1ec1a3a9f54a65598509
+size 302425
diff --git a/wids/tabnet_wids_tablegan.model/args.json b/wids/tabnet_wids_tablegan.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..8b34de182ea8063dc6d63880525d6c53a2e2d8be
--- /dev/null
+++ b/wids/tabnet_wids_tablegan.model/args.json
@@ -0,0 +1,37 @@
+{
+    "name": "tabnet",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_d": 15,
+    "n_steps": 4,
+    "gamma": 1.5764779589451128,
+    "cat_emb_dim": 3,
+    "n_independent": 3,
+    "n_shared": 3,
+    "momentum": 0.04967611500740168,
+    "mask_type": "entmax",
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "tabnet",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabnet_wids_tablegan.model/scaler.json b/wids/tabnet_wids_tablegan.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/tabnet_wids_tablegan.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/tabnet_wids_tablegan.model/weights.pt b/wids/tabnet_wids_tablegan.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..9f9b845085a661a5f894e2d81aeeb52d39989fff
--- /dev/null
+++ b/wids/tabnet_wids_tablegan.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b40aec0c9d2a7c5e09070447dcfb4c71aa2cc7fdffd424aeff1381c10b4c3a02
+size 302425
diff --git a/wids/tabnet_wids_tablegan_madry.model/args.json b/wids/tabnet_wids_tablegan_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..8b34de182ea8063dc6d63880525d6c53a2e2d8be
--- /dev/null
+++ b/wids/tabnet_wids_tablegan_madry.model/args.json
@@ -0,0 +1,37 @@
+{
+    "name": "tabnet",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_d": 15,
+    "n_steps": 4,
+    "gamma": 1.5764779589451128,
+    "cat_emb_dim": 3,
+    "n_independent": 3,
+    "n_shared": 3,
+    "momentum": 0.04967611500740168,
+    "mask_type": "entmax",
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "tabnet",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabnet_wids_tablegan_madry.model/scaler.json b/wids/tabnet_wids_tablegan_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/tabnet_wids_tablegan_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/tabnet_wids_tablegan_madry.model/weights.pt b/wids/tabnet_wids_tablegan_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..2cc3e0465ed3a3178903f1da6478e925623c33f1
--- /dev/null
+++ b/wids/tabnet_wids_tablegan_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b151591b8e6d03ea415c4390e2e9c1e54d2bbd71cd08f4e578909254b85d4203
+size 302425
diff --git a/wids/tabnet_wids_tvae.model/args.json b/wids/tabnet_wids_tvae.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..8b34de182ea8063dc6d63880525d6c53a2e2d8be
--- /dev/null
+++ b/wids/tabnet_wids_tvae.model/args.json
@@ -0,0 +1,37 @@
+{
+    "name": "tabnet",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_d": 15,
+    "n_steps": 4,
+    "gamma": 1.5764779589451128,
+    "cat_emb_dim": 3,
+    "n_independent": 3,
+    "n_shared": 3,
+    "momentum": 0.04967611500740168,
+    "mask_type": "entmax",
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "tabnet",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabnet_wids_tvae.model/scaler.json b/wids/tabnet_wids_tvae.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/tabnet_wids_tvae.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/tabnet_wids_tvae.model/weights.pt b/wids/tabnet_wids_tvae.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..cddc124097214240470fbccb9a14595b643affd6
--- /dev/null
+++ b/wids/tabnet_wids_tvae.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:89eb8f0b74e56575bce90afdfd3f39ea3151fc659083955f2122b2a28292e1b6
+size 302425
diff --git a/wids/tabnet_wids_tvae_madry.model/args.json b/wids/tabnet_wids_tvae_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..8b34de182ea8063dc6d63880525d6c53a2e2d8be
--- /dev/null
+++ b/wids/tabnet_wids_tvae_madry.model/args.json
@@ -0,0 +1,37 @@
+{
+    "name": "tabnet",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_d": 15,
+    "n_steps": 4,
+    "gamma": 1.5764779589451128,
+    "cat_emb_dim": 3,
+    "n_independent": 3,
+    "n_shared": 3,
+    "momentum": 0.04967611500740168,
+    "mask_type": "entmax",
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "tabnet",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabnet_wids_tvae_madry.model/scaler.json b/wids/tabnet_wids_tvae_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/tabnet_wids_tvae_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/tabnet_wids_tvae_madry.model/weights.pt b/wids/tabnet_wids_tvae_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..b3bf157b76b77e25d4c88c605234da6fc8dd76d5
--- /dev/null
+++ b/wids/tabnet_wids_tvae_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b64b08d3cf3d7d7c1f9909bc5cd9a3e94a02e5c1b1cccf84c645ba84b1207cac
+size 302425
diff --git a/wids/tabnet_wids_wgan.model/args.json b/wids/tabnet_wids_wgan.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..8b34de182ea8063dc6d63880525d6c53a2e2d8be
--- /dev/null
+++ b/wids/tabnet_wids_wgan.model/args.json
@@ -0,0 +1,37 @@
+{
+    "name": "tabnet",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_d": 15,
+    "n_steps": 4,
+    "gamma": 1.5764779589451128,
+    "cat_emb_dim": 3,
+    "n_independent": 3,
+    "n_shared": 3,
+    "momentum": 0.04967611500740168,
+    "mask_type": "entmax",
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "tabnet",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabnet_wids_wgan.model/scaler.json b/wids/tabnet_wids_wgan.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/tabnet_wids_wgan.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/tabnet_wids_wgan.model/weights.pt b/wids/tabnet_wids_wgan.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..195c418df64f4600bc93d6339d0df7b52be90822
--- /dev/null
+++ b/wids/tabnet_wids_wgan.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:76bdc1b048abed9ba4ed5c60d7ba42d37f4debec08a62acfd642f565731af629
+size 302425
diff --git a/wids/tabnet_wids_wgan_madry.model/args.json b/wids/tabnet_wids_wgan_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..8b34de182ea8063dc6d63880525d6c53a2e2d8be
--- /dev/null
+++ b/wids/tabnet_wids_wgan_madry.model/args.json
@@ -0,0 +1,37 @@
+{
+    "name": "tabnet",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_d": 15,
+    "n_steps": 4,
+    "gamma": 1.5764779589451128,
+    "cat_emb_dim": 3,
+    "n_independent": 3,
+    "n_shared": 3,
+    "momentum": 0.04967611500740168,
+    "mask_type": "entmax",
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "tabnet",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabnet_wids_wgan_madry.model/scaler.json b/wids/tabnet_wids_wgan_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/tabnet_wids_wgan_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/tabnet_wids_wgan_madry.model/weights.pt b/wids/tabnet_wids_wgan_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..08be833457f419c9468a69b27d6010ac19b3939a
--- /dev/null
+++ b/wids/tabnet_wids_wgan_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f277c8c3945e539fffd5f5a9fc5723bcc63e412c271b1387789d44d669f27e1e
+size 302425
diff --git a/wids/tabtransformer_wids_ctgan.model/args.json b/wids/tabtransformer_wids_ctgan.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..4fb4e8adb5eae378b7a16b1fde7501fc27ec9af6
--- /dev/null
+++ b/wids/tabtransformer_wids_ctgan.model/args.json
@@ -0,0 +1,30 @@
+{
+    "name": "tabtransformer",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -4,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "dim": 32,
+    "depth": 1,
+    "heads": 8,
+    "dropout": 0.2,
+    "custom_dataloader": "",
+    "model_name": "tabtransformer",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabtransformer_wids_ctgan.model/weights.pt b/wids/tabtransformer_wids_ctgan.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..91fc5093b5b292f23c26ea71967c23f2b4f63443
--- /dev/null
+++ b/wids/tabtransformer_wids_ctgan.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:52a5d12bf159900eb5b8d3e48d5d9d297760a747dcba4efcec154f9f11d82745
+size 521062
diff --git a/wids/tabtransformer_wids_ctgan_madry.model/args.json b/wids/tabtransformer_wids_ctgan_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..4fb4e8adb5eae378b7a16b1fde7501fc27ec9af6
--- /dev/null
+++ b/wids/tabtransformer_wids_ctgan_madry.model/args.json
@@ -0,0 +1,30 @@
+{
+    "name": "tabtransformer",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -4,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "dim": 32,
+    "depth": 1,
+    "heads": 8,
+    "dropout": 0.2,
+    "custom_dataloader": "",
+    "model_name": "tabtransformer",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabtransformer_wids_ctgan_madry.model/weights.pt b/wids/tabtransformer_wids_ctgan_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..67a07b42be80215f445abf5902f6123abded5ee9
--- /dev/null
+++ b/wids/tabtransformer_wids_ctgan_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b6a62ed7b4e9e5b63fd4a88b758865043e277a1ad0706d5af9a6729b515289ac
+size 521062
diff --git a/wids/tabtransformer_wids_cutmix.model/args.json b/wids/tabtransformer_wids_cutmix.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..4fb4e8adb5eae378b7a16b1fde7501fc27ec9af6
--- /dev/null
+++ b/wids/tabtransformer_wids_cutmix.model/args.json
@@ -0,0 +1,30 @@
+{
+    "name": "tabtransformer",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -4,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "dim": 32,
+    "depth": 1,
+    "heads": 8,
+    "dropout": 0.2,
+    "custom_dataloader": "",
+    "model_name": "tabtransformer",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabtransformer_wids_cutmix.model/weights.pt b/wids/tabtransformer_wids_cutmix.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..b22c69e6b35aebb56012c3d09c07decb56cd8eac
--- /dev/null
+++ b/wids/tabtransformer_wids_cutmix.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c0049a8b2dbc75c85b75bc3fe0b2d76929c1d9802b1c2862ad1cbd37536242e4
+size 521062
diff --git a/wids/tabtransformer_wids_cutmix_madry.model/args.json b/wids/tabtransformer_wids_cutmix_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..4fb4e8adb5eae378b7a16b1fde7501fc27ec9af6
--- /dev/null
+++ b/wids/tabtransformer_wids_cutmix_madry.model/args.json
@@ -0,0 +1,30 @@
+{
+    "name": "tabtransformer",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -4,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "dim": 32,
+    "depth": 1,
+    "heads": 8,
+    "dropout": 0.2,
+    "custom_dataloader": "",
+    "model_name": "tabtransformer",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabtransformer_wids_cutmix_madry.model/weights.pt b/wids/tabtransformer_wids_cutmix_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..0ef6474a2cde3a43993e6461475c75bca9338f6e
--- /dev/null
+++ b/wids/tabtransformer_wids_cutmix_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6ffa1cdc4ae6be69dfa32daa2f8aa908a05be5710140c21981b00b4febd00a88
+size 521062
diff --git a/wids/tabtransformer_wids_default.model/args.json b/wids/tabtransformer_wids_default.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..57b9cf33247c357172d256a0448c7d41b2024d81
--- /dev/null
+++ b/wids/tabtransformer_wids_default.model/args.json
@@ -0,0 +1,27 @@
+{
+    "name": "tabtransformer",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -4,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "num_classes": 2,
+    "dim": 32,
+    "depth": 1,
+    "heads": 8,
+    "dropout": 0.2,
+    "custom_dataloader": "",
+    "model_name": "tabtransformer",
+    "dataset": "wids",
+    "num_splits": 5,
+    "seed": 42,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabtransformer_wids_default.model/weights.pt b/wids/tabtransformer_wids_default.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..d421c9c17bade7c6f5d9462ed173af95e774ad15
--- /dev/null
+++ b/wids/tabtransformer_wids_default.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6ef203fa7f7e9d8eeae46ef2e0f2005228ae23a39f4c60ec921c44d7400a0bb2
+size 521062
diff --git a/wids/tabtransformer_wids_dist.model/args.json b/wids/tabtransformer_wids_dist.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..57b9cf33247c357172d256a0448c7d41b2024d81
--- /dev/null
+++ b/wids/tabtransformer_wids_dist.model/args.json
@@ -0,0 +1,27 @@
+{
+    "name": "tabtransformer",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -4,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "num_classes": 2,
+    "dim": 32,
+    "depth": 1,
+    "heads": 8,
+    "dropout": 0.2,
+    "custom_dataloader": "",
+    "model_name": "tabtransformer",
+    "dataset": "wids",
+    "num_splits": 5,
+    "seed": 42,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabtransformer_wids_dist.model/weights.pt b/wids/tabtransformer_wids_dist.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..1b41e28a2d1fe7a25501c981a0c180ac885ec2de
--- /dev/null
+++ b/wids/tabtransformer_wids_dist.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:80d860e1e4b1a69e331214becb5e242df4035460f5654156c47efe8e6a127304
+size 521062
diff --git a/wids/tabtransformer_wids_goggle.model/args.json b/wids/tabtransformer_wids_goggle.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..4fb4e8adb5eae378b7a16b1fde7501fc27ec9af6
--- /dev/null
+++ b/wids/tabtransformer_wids_goggle.model/args.json
@@ -0,0 +1,30 @@
+{
+    "name": "tabtransformer",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -4,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "dim": 32,
+    "depth": 1,
+    "heads": 8,
+    "dropout": 0.2,
+    "custom_dataloader": "",
+    "model_name": "tabtransformer",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabtransformer_wids_goggle.model/weights.pt b/wids/tabtransformer_wids_goggle.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..6a010270dc052201ff7a897e570a53d330c3acb0
--- /dev/null
+++ b/wids/tabtransformer_wids_goggle.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:89bb880d888171bfcf2cb9ff2cbdad527cdef17aefd4d335ece800d99b06101c
+size 521062
diff --git a/wids/tabtransformer_wids_goggle_madry.model/args.json b/wids/tabtransformer_wids_goggle_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..4fb4e8adb5eae378b7a16b1fde7501fc27ec9af6
--- /dev/null
+++ b/wids/tabtransformer_wids_goggle_madry.model/args.json
@@ -0,0 +1,30 @@
+{
+    "name": "tabtransformer",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -4,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "dim": 32,
+    "depth": 1,
+    "heads": 8,
+    "dropout": 0.2,
+    "custom_dataloader": "",
+    "model_name": "tabtransformer",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabtransformer_wids_goggle_madry.model/weights.pt b/wids/tabtransformer_wids_goggle_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..52209081ae31d87f9bfa3841cf45569b0775c247
--- /dev/null
+++ b/wids/tabtransformer_wids_goggle_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d559e6507460ddfd32e3cb51a89be6b217cc477b65cf96f900b593e6d05e1ec7
+size 521062
diff --git a/wids/tabtransformer_wids_madry.model/args.json b/wids/tabtransformer_wids_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..57b9cf33247c357172d256a0448c7d41b2024d81
--- /dev/null
+++ b/wids/tabtransformer_wids_madry.model/args.json
@@ -0,0 +1,27 @@
+{
+    "name": "tabtransformer",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -4,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "num_classes": 2,
+    "dim": 32,
+    "depth": 1,
+    "heads": 8,
+    "dropout": 0.2,
+    "custom_dataloader": "",
+    "model_name": "tabtransformer",
+    "dataset": "wids",
+    "num_splits": 5,
+    "seed": 42,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabtransformer_wids_madry.model/weights.pt b/wids/tabtransformer_wids_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..d253a3742005e18d0e579fd9b1b5edc854f8da4e
--- /dev/null
+++ b/wids/tabtransformer_wids_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1cbfcc5bb5a02903762923a9a2f9a42ca0df2ccb20717dbc2e9e3b9aeca312fa
+size 521062
diff --git a/wids/tabtransformer_wids_subset.model/args.json b/wids/tabtransformer_wids_subset.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..57b9cf33247c357172d256a0448c7d41b2024d81
--- /dev/null
+++ b/wids/tabtransformer_wids_subset.model/args.json
@@ -0,0 +1,27 @@
+{
+    "name": "tabtransformer",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -4,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "num_classes": 2,
+    "dim": 32,
+    "depth": 1,
+    "heads": 8,
+    "dropout": 0.2,
+    "custom_dataloader": "",
+    "model_name": "tabtransformer",
+    "dataset": "wids",
+    "num_splits": 5,
+    "seed": 42,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabtransformer_wids_subset.model/weights.pt b/wids/tabtransformer_wids_subset.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..23739b6b26da81ef490b9e64978e5be9c34e08ad
--- /dev/null
+++ b/wids/tabtransformer_wids_subset.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:00a0d8342217a836e433a39d532a5ae2c0316d6d9f9420a20ab03f483b51c2da
+size 521062
diff --git a/wids/tabtransformer_wids_tablegan.model/args.json b/wids/tabtransformer_wids_tablegan.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..4fb4e8adb5eae378b7a16b1fde7501fc27ec9af6
--- /dev/null
+++ b/wids/tabtransformer_wids_tablegan.model/args.json
@@ -0,0 +1,30 @@
+{
+    "name": "tabtransformer",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -4,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "dim": 32,
+    "depth": 1,
+    "heads": 8,
+    "dropout": 0.2,
+    "custom_dataloader": "",
+    "model_name": "tabtransformer",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabtransformer_wids_tablegan.model/weights.pt b/wids/tabtransformer_wids_tablegan.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..f2862216acb6b69c5b9ce8f5e9ceaad7f818a8c5
--- /dev/null
+++ b/wids/tabtransformer_wids_tablegan.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d64339a25a714e73c71ef4915560ab47fdcea8692a6e8878cdc639184dfef0a0
+size 521062
diff --git a/wids/tabtransformer_wids_tablegan_madry.model/args.json b/wids/tabtransformer_wids_tablegan_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..4fb4e8adb5eae378b7a16b1fde7501fc27ec9af6
--- /dev/null
+++ b/wids/tabtransformer_wids_tablegan_madry.model/args.json
@@ -0,0 +1,30 @@
+{
+    "name": "tabtransformer",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -4,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "dim": 32,
+    "depth": 1,
+    "heads": 8,
+    "dropout": 0.2,
+    "custom_dataloader": "",
+    "model_name": "tabtransformer",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabtransformer_wids_tablegan_madry.model/weights.pt b/wids/tabtransformer_wids_tablegan_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..6353b04b316a792d25ca7670169a49ef0d2ee98c
--- /dev/null
+++ b/wids/tabtransformer_wids_tablegan_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:523c825bffedbec8b42cbc91e5c8b5536aac88b3eae09e1193b4c7d7a66df9d0
+size 521062
diff --git a/wids/tabtransformer_wids_tvae.model/args.json b/wids/tabtransformer_wids_tvae.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..4fb4e8adb5eae378b7a16b1fde7501fc27ec9af6
--- /dev/null
+++ b/wids/tabtransformer_wids_tvae.model/args.json
@@ -0,0 +1,30 @@
+{
+    "name": "tabtransformer",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -4,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "dim": 32,
+    "depth": 1,
+    "heads": 8,
+    "dropout": 0.2,
+    "custom_dataloader": "",
+    "model_name": "tabtransformer",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabtransformer_wids_tvae.model/weights.pt b/wids/tabtransformer_wids_tvae.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..66b6695e189c0824399af02b61ce43535c745d66
--- /dev/null
+++ b/wids/tabtransformer_wids_tvae.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:201b50e3d68f3a816d55e401cc66c4d4979d669d29684d1167672f90d10631a0
+size 521062
diff --git a/wids/tabtransformer_wids_tvae_madry.model/args.json b/wids/tabtransformer_wids_tvae_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..4fb4e8adb5eae378b7a16b1fde7501fc27ec9af6
--- /dev/null
+++ b/wids/tabtransformer_wids_tvae_madry.model/args.json
@@ -0,0 +1,30 @@
+{
+    "name": "tabtransformer",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -4,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "dim": 32,
+    "depth": 1,
+    "heads": 8,
+    "dropout": 0.2,
+    "custom_dataloader": "",
+    "model_name": "tabtransformer",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabtransformer_wids_tvae_madry.model/weights.pt b/wids/tabtransformer_wids_tvae_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..9ac71157d548ba0e6def4bedf0fbec8426630f6d
--- /dev/null
+++ b/wids/tabtransformer_wids_tvae_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:43b3c8177f42781deac53c097072ebd0c91476a3b5c5f8111e9fb21069eeae57
+size 521062
diff --git a/wids/tabtransformer_wids_wgan.model/args.json b/wids/tabtransformer_wids_wgan.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..4fb4e8adb5eae378b7a16b1fde7501fc27ec9af6
--- /dev/null
+++ b/wids/tabtransformer_wids_wgan.model/args.json
@@ -0,0 +1,30 @@
+{
+    "name": "tabtransformer",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -4,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "dim": 32,
+    "depth": 1,
+    "heads": 8,
+    "dropout": 0.2,
+    "custom_dataloader": "",
+    "model_name": "tabtransformer",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabtransformer_wids_wgan.model/weights.pt b/wids/tabtransformer_wids_wgan.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..5ec7adcab9bc1de15722777aa82001889e96f3d0
--- /dev/null
+++ b/wids/tabtransformer_wids_wgan.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7c7ee6ceba6dcd223be9a366595782eda5d3c37b73382d44bbb53cd355589ef3
+size 521062
diff --git a/wids/tabtransformer_wids_wgan_madry.model/args.json b/wids/tabtransformer_wids_wgan_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..4fb4e8adb5eae378b7a16b1fde7501fc27ec9af6
--- /dev/null
+++ b/wids/tabtransformer_wids_wgan_madry.model/args.json
@@ -0,0 +1,30 @@
+{
+    "name": "tabtransformer",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -4,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "dim": 32,
+    "depth": 1,
+    "heads": 8,
+    "dropout": 0.2,
+    "custom_dataloader": "",
+    "model_name": "tabtransformer",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/tabtransformer_wids_wgan_madry.model/weights.pt b/wids/tabtransformer_wids_wgan_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..2362cfce2f5cff7d59cecad3ccafa202e3390733
--- /dev/null
+++ b/wids/tabtransformer_wids_wgan_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2066f9796e46ae44beee6d590b268e47d4fba5ad2c98f348107254b7174a2aa1
+size 521062
diff --git a/wids/torchrln_wids_ctgan.model/args.json b/wids/torchrln_wids_ctgan.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..7e509d336ede08db4de639e88848eb60317a1019
--- /dev/null
+++ b/wids/torchrln_wids_ctgan.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "torchrln",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.0006264422188924025,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": 0,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_layers": 2,
+    "hidden_dim": 20,
+    "norm": 2,
+    "theta": -8,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "torchrln",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/torchrln_wids_ctgan.model/scaler.json b/wids/torchrln_wids_ctgan.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/torchrln_wids_ctgan.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/torchrln_wids_ctgan.model/weights.pt b/wids/torchrln_wids_ctgan.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..1ebee5936df97226e1d15fd8ca418aac24912d6c
--- /dev/null
+++ b/wids/torchrln_wids_ctgan.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:023685ceb556cd646539b434b190ef6e0adae78420eb33143553ff25f7894116
+size 25943
diff --git a/wids/torchrln_wids_ctgan_madry.model/args.json b/wids/torchrln_wids_ctgan_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..7e509d336ede08db4de639e88848eb60317a1019
--- /dev/null
+++ b/wids/torchrln_wids_ctgan_madry.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "torchrln",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.0006264422188924025,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": 0,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_layers": 2,
+    "hidden_dim": 20,
+    "norm": 2,
+    "theta": -8,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "torchrln",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/torchrln_wids_ctgan_madry.model/scaler.json b/wids/torchrln_wids_ctgan_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/torchrln_wids_ctgan_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/torchrln_wids_ctgan_madry.model/weights.pt b/wids/torchrln_wids_ctgan_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..97627dd1da1c9cf2e3de95848627352e2e1eed85
--- /dev/null
+++ b/wids/torchrln_wids_ctgan_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0f37137d6961a8b92488bf6735ece8a6425768a1e04358a554efbe7e852af80d
+size 25943
diff --git a/wids/torchrln_wids_cutmix.model/args.json b/wids/torchrln_wids_cutmix.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..7e509d336ede08db4de639e88848eb60317a1019
--- /dev/null
+++ b/wids/torchrln_wids_cutmix.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "torchrln",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.0006264422188924025,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": 0,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_layers": 2,
+    "hidden_dim": 20,
+    "norm": 2,
+    "theta": -8,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "torchrln",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/torchrln_wids_cutmix.model/scaler.json b/wids/torchrln_wids_cutmix.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/torchrln_wids_cutmix.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/torchrln_wids_cutmix.model/weights.pt b/wids/torchrln_wids_cutmix.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..3931094b0952612567a7400b132d54191b3e87d2
--- /dev/null
+++ b/wids/torchrln_wids_cutmix.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fec71a2f0892ffffe5d9bc98e69c3eca8ad0a78bf7152fd2eaf7b783b42a6fef
+size 25943
diff --git a/wids/torchrln_wids_cutmix_madry.model/args.json b/wids/torchrln_wids_cutmix_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..7e509d336ede08db4de639e88848eb60317a1019
--- /dev/null
+++ b/wids/torchrln_wids_cutmix_madry.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "torchrln",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.0006264422188924025,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": 0,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_layers": 2,
+    "hidden_dim": 20,
+    "norm": 2,
+    "theta": -8,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "torchrln",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/torchrln_wids_cutmix_madry.model/scaler.json b/wids/torchrln_wids_cutmix_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/torchrln_wids_cutmix_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/torchrln_wids_cutmix_madry.model/weights.pt b/wids/torchrln_wids_cutmix_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..efb48a174d3210ca1023206549545a188675cae0
--- /dev/null
+++ b/wids/torchrln_wids_cutmix_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fac1fa2a7ebef1ffd4d7267b165651d700a1c713ceb0f27fd7de48195a911ec1
+size 25943
diff --git a/wids/torchrln_wids_default.model/args.json b/wids/torchrln_wids_default.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..8357d1df9b8e7caf892568da832cb51bf24c287d
--- /dev/null
+++ b/wids/torchrln_wids_default.model/args.json
@@ -0,0 +1,30 @@
+{
+    "name": "torchrln",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.0006264422188924025,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": 0,
+    "force_device": "cpu",
+    "is_text": false,
+    "num_classes": 2,
+    "n_layers": 2,
+    "hidden_dim": 20,
+    "norm": 2,
+    "theta": -8,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "torchrln",
+    "dataset": "wids",
+    "num_splits": 5,
+    "seed": 42,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/torchrln_wids_default.model/weights.pt b/wids/torchrln_wids_default.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..3b31fa25e551bf7dfead3281c537c9b9ee3a7e7a
--- /dev/null
+++ b/wids/torchrln_wids_default.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a02917a1c46988fb20c03a957a0f83192b1bc5b1dc54888f56c8a5ffa7f91231
+size 25943
diff --git a/wids/torchrln_wids_dist.model/args.json b/wids/torchrln_wids_dist.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..8357d1df9b8e7caf892568da832cb51bf24c287d
--- /dev/null
+++ b/wids/torchrln_wids_dist.model/args.json
@@ -0,0 +1,30 @@
+{
+    "name": "torchrln",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.0006264422188924025,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": 0,
+    "force_device": "cpu",
+    "is_text": false,
+    "num_classes": 2,
+    "n_layers": 2,
+    "hidden_dim": 20,
+    "norm": 2,
+    "theta": -8,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "torchrln",
+    "dataset": "wids",
+    "num_splits": 5,
+    "seed": 42,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/torchrln_wids_dist.model/weights.pt b/wids/torchrln_wids_dist.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..6d0734b1d02ec667830bbef85613ef1018f955f2
--- /dev/null
+++ b/wids/torchrln_wids_dist.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fd38e9c9f30cf51bf8b583886e44cd3e8438b8167fa54983ef2068735e12a74f
+size 25943
diff --git a/wids/torchrln_wids_goggle.model/args.json b/wids/torchrln_wids_goggle.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..7e509d336ede08db4de639e88848eb60317a1019
--- /dev/null
+++ b/wids/torchrln_wids_goggle.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "torchrln",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.0006264422188924025,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": 0,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_layers": 2,
+    "hidden_dim": 20,
+    "norm": 2,
+    "theta": -8,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "torchrln",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/torchrln_wids_goggle.model/scaler.json b/wids/torchrln_wids_goggle.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/torchrln_wids_goggle.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/torchrln_wids_goggle.model/weights.pt b/wids/torchrln_wids_goggle.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..9fd1f7c2677f05ca34827a75a6fab702ff8c7812
--- /dev/null
+++ b/wids/torchrln_wids_goggle.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0385e5cc30e03c7fa2e24a6bb55f87b4e13d77c528d2bb9b8c5fd67731c2696d
+size 25943
diff --git a/wids/torchrln_wids_goggle_madry.model/args.json b/wids/torchrln_wids_goggle_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..7e509d336ede08db4de639e88848eb60317a1019
--- /dev/null
+++ b/wids/torchrln_wids_goggle_madry.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "torchrln",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.0006264422188924025,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": 0,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_layers": 2,
+    "hidden_dim": 20,
+    "norm": 2,
+    "theta": -8,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "torchrln",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/torchrln_wids_goggle_madry.model/scaler.json b/wids/torchrln_wids_goggle_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/torchrln_wids_goggle_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/torchrln_wids_goggle_madry.model/weights.pt b/wids/torchrln_wids_goggle_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..21404e5f9edb6d36d4068a0ab551caebe27d6093
--- /dev/null
+++ b/wids/torchrln_wids_goggle_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a85c75cd68b7e621f8727ab2c22c18b36a4da94ceb30af0e294333b20d676915
+size 25943
diff --git a/wids/torchrln_wids_madry.model/args.json b/wids/torchrln_wids_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..8357d1df9b8e7caf892568da832cb51bf24c287d
--- /dev/null
+++ b/wids/torchrln_wids_madry.model/args.json
@@ -0,0 +1,30 @@
+{
+    "name": "torchrln",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.0006264422188924025,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": 0,
+    "force_device": "cpu",
+    "is_text": false,
+    "num_classes": 2,
+    "n_layers": 2,
+    "hidden_dim": 20,
+    "norm": 2,
+    "theta": -8,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "torchrln",
+    "dataset": "wids",
+    "num_splits": 5,
+    "seed": 42,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/torchrln_wids_madry.model/weights.pt b/wids/torchrln_wids_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..9b5e1d13e20c0d82fffdbb0b03176fb79003f7e0
--- /dev/null
+++ b/wids/torchrln_wids_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:b4503e6c3f2ba6adbae3c76dc7a452f7f9fcc5c481ee239d49a0cd80ffa1330a
+size 25943
diff --git a/wids/torchrln_wids_subset.model/args.json b/wids/torchrln_wids_subset.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..8357d1df9b8e7caf892568da832cb51bf24c287d
--- /dev/null
+++ b/wids/torchrln_wids_subset.model/args.json
@@ -0,0 +1,30 @@
+{
+    "name": "torchrln",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.0006264422188924025,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": 0,
+    "force_device": "cpu",
+    "is_text": false,
+    "num_classes": 2,
+    "n_layers": 2,
+    "hidden_dim": 20,
+    "norm": 2,
+    "theta": -8,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "torchrln",
+    "dataset": "wids",
+    "num_splits": 5,
+    "seed": 42,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/torchrln_wids_subset.model/weights.pt b/wids/torchrln_wids_subset.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..76f06fa68ba57c21d4a6ff527190dbf904e4ce7e
--- /dev/null
+++ b/wids/torchrln_wids_subset.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:78f4300ee7cb7ce205fbbcf47a3807ab4ad50755a04c94a8ce826af013c5f4b5
+size 25943
diff --git a/wids/torchrln_wids_tablegan.model/args.json b/wids/torchrln_wids_tablegan.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..7e509d336ede08db4de639e88848eb60317a1019
--- /dev/null
+++ b/wids/torchrln_wids_tablegan.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "torchrln",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.0006264422188924025,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": 0,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_layers": 2,
+    "hidden_dim": 20,
+    "norm": 2,
+    "theta": -8,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "torchrln",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/torchrln_wids_tablegan.model/scaler.json b/wids/torchrln_wids_tablegan.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/torchrln_wids_tablegan.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/torchrln_wids_tablegan.model/weights.pt b/wids/torchrln_wids_tablegan.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..90a76eefdacde36ccebed732bd5effa05172e11e
--- /dev/null
+++ b/wids/torchrln_wids_tablegan.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:27e8a4f6a87217a677f83af4266d9a1b372d92704d2e5134cb7581e3a96cff22
+size 25943
diff --git a/wids/torchrln_wids_tablegan_madry.model/args.json b/wids/torchrln_wids_tablegan_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..7e509d336ede08db4de639e88848eb60317a1019
--- /dev/null
+++ b/wids/torchrln_wids_tablegan_madry.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "torchrln",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.0006264422188924025,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": 0,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_layers": 2,
+    "hidden_dim": 20,
+    "norm": 2,
+    "theta": -8,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "torchrln",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/torchrln_wids_tablegan_madry.model/scaler.json b/wids/torchrln_wids_tablegan_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/torchrln_wids_tablegan_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/torchrln_wids_tablegan_madry.model/weights.pt b/wids/torchrln_wids_tablegan_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..79f107b1a4e1d30224fa09d6d15ab08c62659351
--- /dev/null
+++ b/wids/torchrln_wids_tablegan_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:25fffe496698d2b0f7322291eb5ca3d05983eb46bef829066acfda1f924d5970
+size 25943
diff --git a/wids/torchrln_wids_tvae.model/args.json b/wids/torchrln_wids_tvae.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..7e509d336ede08db4de639e88848eb60317a1019
--- /dev/null
+++ b/wids/torchrln_wids_tvae.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "torchrln",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.0006264422188924025,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": 0,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_layers": 2,
+    "hidden_dim": 20,
+    "norm": 2,
+    "theta": -8,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "torchrln",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/torchrln_wids_tvae.model/scaler.json b/wids/torchrln_wids_tvae.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/torchrln_wids_tvae.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/torchrln_wids_tvae.model/weights.pt b/wids/torchrln_wids_tvae.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..109c5fd28b5e1800df0fd3a232a44f6ccb76d0a5
--- /dev/null
+++ b/wids/torchrln_wids_tvae.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bcbcea1282a33afa45679436ba8052e4ec331210a7e21c3f511e2486d4b35f6a
+size 25943
diff --git a/wids/torchrln_wids_tvae_madry.model/args.json b/wids/torchrln_wids_tvae_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..7e509d336ede08db4de639e88848eb60317a1019
--- /dev/null
+++ b/wids/torchrln_wids_tvae_madry.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "torchrln",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.0006264422188924025,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": 0,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_layers": 2,
+    "hidden_dim": 20,
+    "norm": 2,
+    "theta": -8,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "torchrln",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/torchrln_wids_tvae_madry.model/scaler.json b/wids/torchrln_wids_tvae_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/torchrln_wids_tvae_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/torchrln_wids_tvae_madry.model/weights.pt b/wids/torchrln_wids_tvae_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..3a98feb6b3b853be2447a8c36bf3aadbbeaaefcb
--- /dev/null
+++ b/wids/torchrln_wids_tvae_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c8dae8af09d38e4760a4ad000a3349420b5b7e30c458f6a4dbef0ca576fa9ad7
+size 25943
diff --git a/wids/torchrln_wids_wgan.model/args.json b/wids/torchrln_wids_wgan.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..7e509d336ede08db4de639e88848eb60317a1019
--- /dev/null
+++ b/wids/torchrln_wids_wgan.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "torchrln",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.0006264422188924025,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": 0,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_layers": 2,
+    "hidden_dim": 20,
+    "norm": 2,
+    "theta": -8,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "torchrln",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/torchrln_wids_wgan.model/scaler.json b/wids/torchrln_wids_wgan.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/torchrln_wids_wgan.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/torchrln_wids_wgan.model/weights.pt b/wids/torchrln_wids_wgan.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..0767db919b0f84a47876f0c90d1623cd97897059
--- /dev/null
+++ b/wids/torchrln_wids_wgan.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0da6fc52dbe65a71de51632aa4d61487f4c37c34abed988b8d2cd2d3f4fad823
+size 25943
diff --git a/wids/torchrln_wids_wgan_madry.model/args.json b/wids/torchrln_wids_wgan_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..7e509d336ede08db4de639e88848eb60317a1019
--- /dev/null
+++ b/wids/torchrln_wids_wgan_madry.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "torchrln",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": 0.0006264422188924025,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": 0,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "n_layers": 2,
+    "hidden_dim": 20,
+    "norm": 2,
+    "theta": -8,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "torchrln",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/torchrln_wids_wgan_madry.model/scaler.json b/wids/torchrln_wids_wgan_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/torchrln_wids_wgan_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/torchrln_wids_wgan_madry.model/weights.pt b/wids/torchrln_wids_wgan_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..2ec600339ff16add6b5e5cb678207ffd778f8d81
--- /dev/null
+++ b/wids/torchrln_wids_wgan_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1f38d0ce808b1aedc53e6790e19d1f2a5138f1799931d35773958a9f4cd3464e
+size 25943
diff --git a/wids/vime_wids_ctgan.model/args.json b/wids/vime_wids_ctgan.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..f7907240abc1b469e9fd721eeb6a83f4e1f5aa6b
--- /dev/null
+++ b/wids/vime_wids_ctgan.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "saint",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "p_m": 0.38262831496807614,
+    "K": 10,
+    "alpha": 0.1308177895044439,
+    "beta": 0.254045304129745,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "vime",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/vime_wids_ctgan.model/scaler.json b/wids/vime_wids_ctgan.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/vime_wids_ctgan.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/vime_wids_ctgan.model/semi_weights.pt b/wids/vime_wids_ctgan.model/semi_weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..9fd72e34c2e4f4ce225b5df2cd2bc2b7366797d9
--- /dev/null
+++ b/wids/vime_wids_ctgan.model/semi_weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:82395caee59a555e607f93e56fa5b0ce9053a5b4da6eed365c4baf54bd6c8b9b
+size 209409
diff --git a/wids/vime_wids_ctgan.model/weights.pt b/wids/vime_wids_ctgan.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..741289a3949f4ef2ad14a2026b88224848c6de81
--- /dev/null
+++ b/wids/vime_wids_ctgan.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8deab1c0f112a794c73e054ff14fd2a34979bba2dccafeaeebc4bff740fa8b68
+size 143127
diff --git a/wids/vime_wids_ctgan_madry.model/args.json b/wids/vime_wids_ctgan_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..f7907240abc1b469e9fd721eeb6a83f4e1f5aa6b
--- /dev/null
+++ b/wids/vime_wids_ctgan_madry.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "saint",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "p_m": 0.38262831496807614,
+    "K": 10,
+    "alpha": 0.1308177895044439,
+    "beta": 0.254045304129745,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "vime",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/vime_wids_ctgan_madry.model/scaler.json b/wids/vime_wids_ctgan_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/vime_wids_ctgan_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/vime_wids_ctgan_madry.model/semi_weights.pt b/wids/vime_wids_ctgan_madry.model/semi_weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..36277f57267189e5cfe522d1be8bd5c55186be63
--- /dev/null
+++ b/wids/vime_wids_ctgan_madry.model/semi_weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a5d9b23acb84cfa6393899ce282d76a22b2880768611afe9dd04f0b30b8571e8
+size 209409
diff --git a/wids/vime_wids_ctgan_madry.model/weights.pt b/wids/vime_wids_ctgan_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..c87f9f30c35432ad09ba31e3da834292713c620e
--- /dev/null
+++ b/wids/vime_wids_ctgan_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4fe404c009ab26a012c82c2d735317dd06c0495763245b4dbdd6aeb6526a17a7
+size 143127
diff --git a/wids/vime_wids_cutmix.model/args.json b/wids/vime_wids_cutmix.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..f7907240abc1b469e9fd721eeb6a83f4e1f5aa6b
--- /dev/null
+++ b/wids/vime_wids_cutmix.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "saint",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "p_m": 0.38262831496807614,
+    "K": 10,
+    "alpha": 0.1308177895044439,
+    "beta": 0.254045304129745,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "vime",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/vime_wids_cutmix.model/scaler.json b/wids/vime_wids_cutmix.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/vime_wids_cutmix.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/vime_wids_cutmix.model/semi_weights.pt b/wids/vime_wids_cutmix.model/semi_weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..3a6f04788fa2faf3b77dcd51ead02fa7d77c7582
--- /dev/null
+++ b/wids/vime_wids_cutmix.model/semi_weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:27a171fba58fb7522b4b5204fff1af4cdcaceae2ca497e190649f5fbfd237865
+size 209409
diff --git a/wids/vime_wids_cutmix.model/weights.pt b/wids/vime_wids_cutmix.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..91fb35da2c10372420a32739fa39c1ee96b67910
--- /dev/null
+++ b/wids/vime_wids_cutmix.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dc1f6fc09553b0f09ba02613977a56f79325504af43078fedd32a5f9f393907c
+size 143127
diff --git a/wids/vime_wids_cutmix_madry.model/args.json b/wids/vime_wids_cutmix_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..f7907240abc1b469e9fd721eeb6a83f4e1f5aa6b
--- /dev/null
+++ b/wids/vime_wids_cutmix_madry.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "saint",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "p_m": 0.38262831496807614,
+    "K": 10,
+    "alpha": 0.1308177895044439,
+    "beta": 0.254045304129745,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "vime",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/vime_wids_cutmix_madry.model/scaler.json b/wids/vime_wids_cutmix_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/vime_wids_cutmix_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/vime_wids_cutmix_madry.model/semi_weights.pt b/wids/vime_wids_cutmix_madry.model/semi_weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..82717534b0af17961a4168d81efcff1ec0fc0cb0
--- /dev/null
+++ b/wids/vime_wids_cutmix_madry.model/semi_weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:377d896a48f783831f7788ce9ab4a8f0ddf48afb28d427cb6eb88aa11a9619be
+size 209409
diff --git a/wids/vime_wids_cutmix_madry.model/weights.pt b/wids/vime_wids_cutmix_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..0a1ad1e38676cd68ce3d6f00de6cc78dfc77a3ac
--- /dev/null
+++ b/wids/vime_wids_cutmix_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4fde3e4401c9a018b611010070116c53fb11552cf760e7249f762ff7ea1f6cb9
+size 143127
diff --git a/wids/vime_wids_default.model/args.json b/wids/vime_wids_default.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..abbf3bf00599ba4e06ef4858a0f868f54135b01f
--- /dev/null
+++ b/wids/vime_wids_default.model/args.json
@@ -0,0 +1,30 @@
+{
+    "name": "saint",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "num_classes": 2,
+    "p_m": 0.38262831496807614,
+    "K": 10,
+    "alpha": 0.1308177895044439,
+    "beta": 0.254045304129745,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "vime",
+    "dataset": "wids",
+    "num_splits": 5,
+    "seed": 42,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/vime_wids_default.model/semi_weights.pt b/wids/vime_wids_default.model/semi_weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..2e9b8bfdf3e2474c40a84440d67c358993cc752b
--- /dev/null
+++ b/wids/vime_wids_default.model/semi_weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:18267fa7765a8995cb99e9f39be09356eed913df8bafc84230e453b98d3549d8
+size 209409
diff --git a/wids/vime_wids_default.model/weights.pt b/wids/vime_wids_default.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..48be86e00c29b2bdfed857e7ad5a4d60ec8fcbbd
--- /dev/null
+++ b/wids/vime_wids_default.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ad9cec0343bb870f3fbce149c70090fea6e23e44f001fd85d2f6c9396a767970
+size 143127
diff --git a/wids/vime_wids_dist.model/args.json b/wids/vime_wids_dist.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..abbf3bf00599ba4e06ef4858a0f868f54135b01f
--- /dev/null
+++ b/wids/vime_wids_dist.model/args.json
@@ -0,0 +1,30 @@
+{
+    "name": "saint",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "num_classes": 2,
+    "p_m": 0.38262831496807614,
+    "K": 10,
+    "alpha": 0.1308177895044439,
+    "beta": 0.254045304129745,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "vime",
+    "dataset": "wids",
+    "num_splits": 5,
+    "seed": 42,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/vime_wids_dist.model/semi_weights.pt b/wids/vime_wids_dist.model/semi_weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..261065cae6ac338007a559e34874adb5553f3893
--- /dev/null
+++ b/wids/vime_wids_dist.model/semi_weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:08796894c61d1da4b60ffeef0f631e3925c5bc3787e6e8b31427ca05fc1388f6
+size 209409
diff --git a/wids/vime_wids_dist.model/weights.pt b/wids/vime_wids_dist.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..710ea5a2e1f0ac2f065a40534ccb48f64d4e7f30
--- /dev/null
+++ b/wids/vime_wids_dist.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:db1ee62d380145308653a3e92cadfd682a43389b6d6cbcab7fd0fee5c15872bb
+size 143127
diff --git a/wids/vime_wids_goggle.model/args.json b/wids/vime_wids_goggle.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..f7907240abc1b469e9fd721eeb6a83f4e1f5aa6b
--- /dev/null
+++ b/wids/vime_wids_goggle.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "saint",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "p_m": 0.38262831496807614,
+    "K": 10,
+    "alpha": 0.1308177895044439,
+    "beta": 0.254045304129745,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "vime",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/vime_wids_goggle.model/scaler.json b/wids/vime_wids_goggle.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/vime_wids_goggle.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/vime_wids_goggle.model/semi_weights.pt b/wids/vime_wids_goggle.model/semi_weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..76641643c6fbd64782e02a5dcea9c33187ebfc4e
--- /dev/null
+++ b/wids/vime_wids_goggle.model/semi_weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bceddb17772f50fd5f8a8b33935193168bc37d553f0717ad242f515d6c6143d2
+size 209409
diff --git a/wids/vime_wids_goggle.model/weights.pt b/wids/vime_wids_goggle.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..517a20a938cbe64425476c4db6e98f328229c9bc
--- /dev/null
+++ b/wids/vime_wids_goggle.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bc1f8b34f6631e4d8260686bc3f70a6725fa8c5a718cbb2371f46f35cb8cdec9
+size 143127
diff --git a/wids/vime_wids_goggle_madry.model/args.json b/wids/vime_wids_goggle_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..f7907240abc1b469e9fd721eeb6a83f4e1f5aa6b
--- /dev/null
+++ b/wids/vime_wids_goggle_madry.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "saint",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "p_m": 0.38262831496807614,
+    "K": 10,
+    "alpha": 0.1308177895044439,
+    "beta": 0.254045304129745,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "vime",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/vime_wids_goggle_madry.model/scaler.json b/wids/vime_wids_goggle_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/vime_wids_goggle_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/vime_wids_goggle_madry.model/semi_weights.pt b/wids/vime_wids_goggle_madry.model/semi_weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..6a56031c0e6cbf9f7bb6d6bc67c20c020652b8bf
--- /dev/null
+++ b/wids/vime_wids_goggle_madry.model/semi_weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:61732a26c5b739063bbbdec71ba114c261bc886de8430f8c02f24df07ba3a5fb
+size 209409
diff --git a/wids/vime_wids_goggle_madry.model/weights.pt b/wids/vime_wids_goggle_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..12a96b75677fe384d6496020d71cd33ce4700d18
--- /dev/null
+++ b/wids/vime_wids_goggle_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:22bf90afe6069f5b1f3f818b3273415736cfe94a4ceeac0a9d9e6ac592698ae5
+size 143127
diff --git a/wids/vime_wids_madry.model/args.json b/wids/vime_wids_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..abbf3bf00599ba4e06ef4858a0f868f54135b01f
--- /dev/null
+++ b/wids/vime_wids_madry.model/args.json
@@ -0,0 +1,30 @@
+{
+    "name": "saint",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "num_classes": 2,
+    "p_m": 0.38262831496807614,
+    "K": 10,
+    "alpha": 0.1308177895044439,
+    "beta": 0.254045304129745,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "vime",
+    "dataset": "wids",
+    "num_splits": 5,
+    "seed": 42,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/vime_wids_madry.model/semi_weights.pt b/wids/vime_wids_madry.model/semi_weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..1fa51ce05d0a18e710a3dd3294364443c20f427d
--- /dev/null
+++ b/wids/vime_wids_madry.model/semi_weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cfb798acbc114b972bc8d0baef96518c713dbc133e2f70f82aa31c1573fecfb6
+size 209409
diff --git a/wids/vime_wids_madry.model/weights.pt b/wids/vime_wids_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..61f1c80c51c0574437bbee94bf13e23d3785b272
--- /dev/null
+++ b/wids/vime_wids_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:692447702e63074017565373b6d59751e74cec7b454f2d4145bf2c49bf258024
+size 143127
diff --git a/wids/vime_wids_subset.model/args.json b/wids/vime_wids_subset.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..abbf3bf00599ba4e06ef4858a0f868f54135b01f
--- /dev/null
+++ b/wids/vime_wids_subset.model/args.json
@@ -0,0 +1,30 @@
+{
+    "name": "saint",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "num_classes": 2,
+    "p_m": 0.38262831496807614,
+    "K": 10,
+    "alpha": 0.1308177895044439,
+    "beta": 0.254045304129745,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "vime",
+    "dataset": "wids",
+    "num_splits": 5,
+    "seed": 42,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/vime_wids_subset.model/semi_weights.pt b/wids/vime_wids_subset.model/semi_weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..218ec048f605821cfa39ce06410da4c425c4865d
--- /dev/null
+++ b/wids/vime_wids_subset.model/semi_weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a0f860296e0a47fc4c9566fdb84cc760a7266c82a0dc0f5a777829bd87f4a132
+size 209409
diff --git a/wids/vime_wids_subset.model/weights.pt b/wids/vime_wids_subset.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..225f83d7b83e5c4f2494840037e99fa22ed66f87
--- /dev/null
+++ b/wids/vime_wids_subset.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8b8a527eccfa46cfc24ca131b1d4118bacdcad6fcc8346c1422295c971ec5f78
+size 143127
diff --git a/wids/vime_wids_tablegan.model/args.json b/wids/vime_wids_tablegan.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..f7907240abc1b469e9fd721eeb6a83f4e1f5aa6b
--- /dev/null
+++ b/wids/vime_wids_tablegan.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "saint",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "p_m": 0.38262831496807614,
+    "K": 10,
+    "alpha": 0.1308177895044439,
+    "beta": 0.254045304129745,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "vime",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/vime_wids_tablegan.model/scaler.json b/wids/vime_wids_tablegan.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/vime_wids_tablegan.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/vime_wids_tablegan.model/semi_weights.pt b/wids/vime_wids_tablegan.model/semi_weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..8cf48ae6724b3d9c9addad204ae28a1699be3a34
--- /dev/null
+++ b/wids/vime_wids_tablegan.model/semi_weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:53e2692e670d0fed4d3af4c2dcc76962c8a364a626b363ffb442830aa85f4f54
+size 209409
diff --git a/wids/vime_wids_tablegan.model/weights.pt b/wids/vime_wids_tablegan.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..fb89858bfb7e3b49ad03b27d44ef57c9d299e0a8
--- /dev/null
+++ b/wids/vime_wids_tablegan.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:33a4c4ec9e024d1e15f059d95207b04b0f6dfd36606db1a3e60b2588aee698bb
+size 143127
diff --git a/wids/vime_wids_tablegan_madry.model/args.json b/wids/vime_wids_tablegan_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..f7907240abc1b469e9fd721eeb6a83f4e1f5aa6b
--- /dev/null
+++ b/wids/vime_wids_tablegan_madry.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "saint",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "p_m": 0.38262831496807614,
+    "K": 10,
+    "alpha": 0.1308177895044439,
+    "beta": 0.254045304129745,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "vime",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/vime_wids_tablegan_madry.model/scaler.json b/wids/vime_wids_tablegan_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/vime_wids_tablegan_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/vime_wids_tablegan_madry.model/semi_weights.pt b/wids/vime_wids_tablegan_madry.model/semi_weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..045e452b85827eaf2fcd66c43ba2cd7c26573bd3
--- /dev/null
+++ b/wids/vime_wids_tablegan_madry.model/semi_weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bc6367f4c9a11d7e8b1e6ceac6348565eae7e46c86947b78c6457c09248656df
+size 209409
diff --git a/wids/vime_wids_tablegan_madry.model/weights.pt b/wids/vime_wids_tablegan_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..c6300dd8a94eb416696ebb61e747044cd92a28ca
--- /dev/null
+++ b/wids/vime_wids_tablegan_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f1e766b4fe2c9e03e989fc81bc5922ca055597a5cd4fc2534942dbef5f07fd48
+size 143127
diff --git a/wids/vime_wids_tvae.model/args.json b/wids/vime_wids_tvae.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..f7907240abc1b469e9fd721eeb6a83f4e1f5aa6b
--- /dev/null
+++ b/wids/vime_wids_tvae.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "saint",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "p_m": 0.38262831496807614,
+    "K": 10,
+    "alpha": 0.1308177895044439,
+    "beta": 0.254045304129745,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "vime",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/vime_wids_tvae.model/scaler.json b/wids/vime_wids_tvae.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/vime_wids_tvae.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/vime_wids_tvae.model/semi_weights.pt b/wids/vime_wids_tvae.model/semi_weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..72bb6152e361724d624a7f7df1db9653e312cab7
--- /dev/null
+++ b/wids/vime_wids_tvae.model/semi_weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:abc2dfe3d86fefef79b6991aacbfd83d4690cbff1ae3e40410956741972a7be9
+size 209409
diff --git a/wids/vime_wids_tvae.model/weights.pt b/wids/vime_wids_tvae.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..afce2030cf576a7eafe0778697e79ba1e6ad74f1
--- /dev/null
+++ b/wids/vime_wids_tvae.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dc0517fc0c5998d8168ca388c6bf81f7ca5c6e05eba5ddf042f3fdcc91cbd449
+size 143127
diff --git a/wids/vime_wids_tvae_madry.model/args.json b/wids/vime_wids_tvae_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..f7907240abc1b469e9fd721eeb6a83f4e1f5aa6b
--- /dev/null
+++ b/wids/vime_wids_tvae_madry.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "saint",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "p_m": 0.38262831496807614,
+    "K": 10,
+    "alpha": 0.1308177895044439,
+    "beta": 0.254045304129745,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "vime",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/vime_wids_tvae_madry.model/scaler.json b/wids/vime_wids_tvae_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/vime_wids_tvae_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/vime_wids_tvae_madry.model/semi_weights.pt b/wids/vime_wids_tvae_madry.model/semi_weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..75ba33077eb432e793bdde348442b6968b873024
--- /dev/null
+++ b/wids/vime_wids_tvae_madry.model/semi_weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:796548b31c9670a665be99ccc886daa800375a67f8c2717e5ed05528ae32e495
+size 209409
diff --git a/wids/vime_wids_tvae_madry.model/weights.pt b/wids/vime_wids_tvae_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..786ff095430eb03df4470691d1d97d987519274d
--- /dev/null
+++ b/wids/vime_wids_tvae_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e073ea2cff160c0335cc0dfa2b3a99b97c98fd00af2aa1235b629d8354f33392
+size 143127
diff --git a/wids/vime_wids_wgan.model/args.json b/wids/vime_wids_wgan.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..f7907240abc1b469e9fd721eeb6a83f4e1f5aa6b
--- /dev/null
+++ b/wids/vime_wids_wgan.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "saint",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "p_m": 0.38262831496807614,
+    "K": 10,
+    "alpha": 0.1308177895044439,
+    "beta": 0.254045304129745,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "vime",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/vime_wids_wgan.model/scaler.json b/wids/vime_wids_wgan.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/vime_wids_wgan.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/vime_wids_wgan.model/semi_weights.pt b/wids/vime_wids_wgan.model/semi_weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..33c911c9b985dd963681e3ff5e1d39fa1710f217
--- /dev/null
+++ b/wids/vime_wids_wgan.model/semi_weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d4c5a1e2a23c0d38b9094aa8b171c870e05068cb14063f3b013ba7a1bb727946
+size 209409
diff --git a/wids/vime_wids_wgan.model/weights.pt b/wids/vime_wids_wgan.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..c4557894320b4016cf901219898f7a7053f1977e
--- /dev/null
+++ b/wids/vime_wids_wgan.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:89ae094e2885c67f76495d253fee6859da4e39a34b2c0fb384073f825900c51b
+size 143127
diff --git a/wids/vime_wids_wgan_madry.model/args.json b/wids/vime_wids_wgan_madry.model/args.json
new file mode 100755
index 0000000000000000000000000000000000000000..f7907240abc1b469e9fd721eeb6a83f4e1f5aa6b
--- /dev/null
+++ b/wids/vime_wids_wgan_madry.model/args.json
@@ -0,0 +1,33 @@
+{
+    "name": "saint",
+    "objective": "classification",
+    "batch_size": 1024,
+    "epochs": 100,
+    "early_stopping_rounds": 100,
+    "learning_rate": -3,
+    "val_batch_size": 2048,
+    "class_weight": "balanced",
+    "weight_decay": -3,
+    "force_device": "cpu",
+    "is_text": false,
+    "dg_method": null,
+    "seed": 42,
+    "dg_num_groups": 4,
+    "dg_group_method": "random",
+    "num_classes": 2,
+    "p_m": 0.38262831496807614,
+    "K": 10,
+    "alpha": 0.1308177895044439,
+    "beta": 0.254045304129745,
+    "custom_dataloader": "",
+    "depth": 12,
+    "heads": 2,
+    "dropout": 0.5,
+    "model_name": "vime",
+    "dataset": "wids",
+    "num_splits": 5,
+    "shuffle": true,
+    "metrics": [
+        "auc"
+    ]
+}
\ No newline at end of file
diff --git a/wids/vime_wids_wgan_madry.model/scaler.json b/wids/vime_wids_wgan_madry.model/scaler.json
new file mode 100755
index 0000000000000000000000000000000000000000..48922988bc967722246b723897f0d94d7a0132d5
--- /dev/null
+++ b/wids/vime_wids_wgan_madry.model/scaler.json
@@ -0,0 +1 @@
+{"x_min": [16.0, 14.844925880432129, 0.0, 137.1999969482422, -0.22430555522441864, 0.0, 38.599998474121094, 101.0, 101.01000213623047, 0.0, 0.0, 4.0, 0.30000001192092896, 1.0, 1.0, 0.0, 1.0, 39.0, 30.0, 16.200000762939453, 0.0, 40.0, 4.0, 117.0, 32.099998474121094, 0.0, 0.8999999761581421, 46.0, 13.0, 46.0, 13.0, 58.0, 0.0, 60.0, 22.0, 60.0, 22.0, 14.0, 0.0, 13.0, 0.0, 90.0, 41.0, 90.0, 41.029998779296875, 35.099998474121094, 31.888999938964844, 37.0, 22.0, 37.0, 22.0, 46.0, 36.0, 49.0, 32.0, 49.0, 32.0, 10.0, 0.0, 10.0, 0.0, 75.0, 53.0, 75.0, 53.0, 33.400001525878906, 32.900001525878906, 4.0, 3.0, 6.199999809265137, 5.5, 0.3400000035762787, 0.30000001192092896, 73.0, 33.0, 12.0, 7.0, 6.800000190734863, 5.300000190734863, 20.399999618530273, 16.100000381469727, 27.0, 18.549999237060547, 2.799999952316284, 2.4000000953674316, 123.0, 117.0, 1.2000000476837158, 0.8999999761581421, -1.0, -1.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0], "x_max": [89.0, 67.81498718261719, 1.0, 195.58999633789062, 67.02361297607422, 0.0, 186.0, 308.0, 2201.050048828125, 1.0, 1.0, 127.0, 11.180000305175781, 4.0, 6.0, 0.0, 5.0, 598.7000122070312, 178.0, 51.400001525878906, 1.0, 200.0, 60.0, 158.0, 39.70000076293945, 1.0, 45.79999923706055, 165.0, 90.0, 165.0, 90.0, 177.0, 143.0, 184.0, 112.0, 181.0, 112.0, 92.0, 72.0, 100.0, 100.0, 232.0, 160.0, 232.0, 160.0, 39.900001525878906, 37.79999923706055, 143.0, 113.0, 144.0, 114.0, 164.0, 144.0, 165.0, 138.0, 163.0, 138.0, 59.0, 118.0, 100.0, 100.0, 223.0, 194.0, 223.0, 195.0, 39.5, 39.29999923706055, 126.0, 113.08999633789062, 10.800000190734863, 10.300000190734863, 11.109999656677246, 9.937899589538574, 611.0, 288.0, 40.0, 39.0, 17.200000762939453, 16.700000762939453, 51.5, 50.0, 585.0, 557.4500122070312, 7.0, 5.800000190734863, 158.0, 153.0, 46.08000183105469, 40.89799880981445, 0.9800000190734863, 0.9700000286102295, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0, 1.0], "x_mean": null, "x_std": null, "categories": [[1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0], [1.0, 2.0, 3.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0], [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0], [2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0, 10.0, 11.0, 12.0, 13.0, 14.0, 15.0, 16.0, 17.0, 18.0, 19.0, 20.0, 21.0, 22.0, 23.0, 24.0, 25.0, 26.0, 27.0, 28.0, 29.0, 30.0, 31.0, 32.0, 33.0, 34.0, 35.0, 36.0, 37.0, 38.0, 39.0, 40.0, 41.0, 42.0, 43.0, 44.0, 45.0, 46.0, 47.0, 48.0, 49.0, 50.0, 51.0, 52.0, 53.0, 54.0, 55.0, 56.0, 57.0, 58.0, 59.0, 60.0, 61.0, 62.0, 63.0, 64.0, 65.0, 66.0, 67.0, 68.0, 69.0, 70.0, 71.0, 72.0, 73.0, 74.0, 75.0, 76.0, 77.0, 78.0, 79.0, 80.0, 81.0, 82.0, 83.0, 84.0, 85.0, 86.0, 87.0, 88.0, 89.0, 90.0, 91.0, 92.0, 93.0, 94.0, 95.0, 96.0, 97.0, 98.0, 99.0, 100.0, 101.0, 102.0, 103.0, 104.0, 105.0, 106.0, 107.0, 108.0, 109.0, 110.0, 111.0, 112.0]], "cat_idx": [3, 4, 6, 7, 8, 9, 105, 106, 107], "num_idx": [0, 1, 2, 5, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 62, 63, 64, 65, 66, 67, 68, 69, 70, 71, 72, 73, 74, 75, 76, 77, 78, 79, 80, 81, 82, 83, 84, 85, 86, 87, 88, 89, 90, 91, 92, 93, 94, 95, 96, 97, 98, 99, 100, 101, 102, 103, 104]}
\ No newline at end of file
diff --git a/wids/vime_wids_wgan_madry.model/semi_weights.pt b/wids/vime_wids_wgan_madry.model/semi_weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..b4eaa7289a83386f012e364fdc997777c865a096
--- /dev/null
+++ b/wids/vime_wids_wgan_madry.model/semi_weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4db22f8b3d78693efb3a0a782c204b4a735ce6ebaecc3a6b98aaebee594ca441
+size 209409
diff --git a/wids/vime_wids_wgan_madry.model/weights.pt b/wids/vime_wids_wgan_madry.model/weights.pt
new file mode 100755
index 0000000000000000000000000000000000000000..413ddb5641d271cc9224fc68a827517324f308b7
--- /dev/null
+++ b/wids/vime_wids_wgan_madry.model/weights.pt
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:54e2db9cee982ce1618e1548c59d5f1f752033f402ba5af24363c2f892664c6a
+size 143127