-
-
Notifications
You must be signed in to change notification settings - Fork 26.2k
Open
Description
Description
I suggest allowing mixed datasets (half binary variables, half real-valued variables) into the Naive Bayes classifier. Currently the GaussianNB
and BernoulliNB
classes handle one case or the other, but not combined. I'd be happy to write the code for this, so I'm curious if this has been explored before and if it would be helpful!
For example, on the Titanic dataset, gender is a bernoulli variable while age is real-valued. Passing both into a Naive Bayes classifier would improve it.
This is related to this currently pending PR: #12569
kylrth, Anders429, jarednielsen, FlorianWilhelm, abhinavsingh23 and 5 more