Skip to content

Commit

Permalink
Fix bool preprocessing (preseve missing as 0)
Browse files Browse the repository at this point in the history
This breaks backwards compatibility with previous models, but
evaluation showed it improves ML-based synthesis performance
significantly.
  • Loading branch information
iburakov committed Aug 23, 2023
1 parent 5927a08 commit 293a7b9
Showing 1 changed file with 1 addition and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -9,8 +9,7 @@


def _map_bool(c):
# TODO: 1 - 0 - -1 mapping
return c.apply(lambda v: 1 if v is True else (0 if v is False else v))
return c.apply(lambda v: 1 if v is True else (-1 if v is False else 0))


def _map_categorical(df, c):
Expand Down

0 comments on commit 293a7b9

Please sign in to comment.