Skip to content

Commit

Permalink
update value mapping; grinsztajn benchmark revalues these to 0/1
Browse files Browse the repository at this point in the history
  • Loading branch information
jpgard committed Jan 11, 2024
1 parent 62c09a7 commit 90a733b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tableshift/datasets/grinsztajn.py
Original file line number Diff line number Diff line change
Expand Up @@ -262,7 +262,7 @@
'Amount of the given credit (NT dollar): it includes both the individual consumer credit and his/her family (supplementary) credit.',
name_extended='total credit amount in NT dollars'),
Feature('x2', int, 'gender', name_extended='gender',
value_mapping={1: 'male', 2: 'female'}),
value_mapping={0: 'male', 1: 'female'}),
Feature('x5', int, 'age', name_extended='age in years'),
Feature('x6', float, name_extended='repayment status in September 2005',
value_mapping=PAYMENT_STATUS_MAPPING),
Expand Down

0 comments on commit 90a733b

Please sign in to comment.