add mappings for more features

mlfoundations · Dec 29, 2023 · 10615e6 · 10615e6
1 parent 3a61d00
commit 10615e6
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 5 deletions.
diff --git a/tableshift/datasets/german.py b/tableshift/datasets/german.py
@@ -102,9 +102,16 @@
                           "applicants",
             value_mapping={
                 1: "greater than or equal to median age",
-                0: "not greater than or equal to median age"
+                '1.0': "greater than or equal to median age",
+                0: "not greater than or equal to median age",
+                '0.0': "not greater than or equal to median age"
             }),
-    Feature("sex", cat_dtype, value_mapping={1: "male", 0: "female"}),
+    Feature("sex", cat_dtype, value_mapping={
+        1: "male",
+        '1.0': "male",
+        0: "female",
+        '0.0': "female",
+    }),
     Feature("other_installment", cat_dtype,
             description="Other installment plans",
             name_extended="Other installment plans",

diff --git a/tableshift/datasets/uci.py b/tableshift/datasets/uci.py
@@ -205,10 +205,10 @@
     Feature('buying', cat_dtype, name_extended='buying price'),
     Feature('maint', cat_dtype, name_extended='maintenance price'),
     Feature('doors', cat_dtype, name_extended='number of doors'),
-    Feature('persons', cat_dtype),
+    Feature('persons', cat_dtype, name_extended='capacity in terms of persons to carry'),
     Feature('lug_boot', cat_dtype,
-            name_extended='capacity in terms of persons'),
-    Feature('safety', cat_dtype, name_extended='safety level'),
+            name_extended='the size of luggage boot'),
+    Feature('safety', cat_dtype, name_extended='estimated safety level of the car'),
     Feature('class', cat_dtype, is_target=True,
             name_extended='car acceptability',
             value_mapping={