Skip to content

Naive Bayes Classifier with Mixed Bernoulli/Gaussian Models #12957

@jarednielsen

Description

@jarednielsen

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

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions