Have OHE return Boolean columns - #2170
Conversation
Codecov Report
@@ Coverage Diff @@
## main #2170 +/- ##
=========================================
+ Coverage 100.0% 100.0% +0.1%
=========================================
Files 295 295
Lines 24376 24389 +13
=========================================
+ Hits 24366 24379 +13
Misses 10 10
Continue to review full report at Codecov.
|
jeremyliweishih
left a comment
There was a problem hiding this comment.
I think this looks great Bryan!
freddyaboulton
left a comment
There was a problem hiding this comment.
Looks good @bchen1116 !!
| if name == 'integers': | ||
| assert str(types) == "Integer" | ||
| else: | ||
| assert str(types) == "Boolean" |
There was a problem hiding this comment.
Looks great! I love that you made this general and not specific to xgboost.
One check I'd add: assert that the right number of features has been generated. That way, if for some reason the OHE isn't generating any features at all for the bool/category inputs, the test will fail.
fix #2169