[IGNITE-9282] [ML] Add Naive Bayes classifier

• Type: Sub-task
• Status: Resolved
• Priority: Major
• Resolution: Done
• Fix Version/s: 2.8
• Component/s: ml
• Labels: new-feature newbie newbie++
• Ignite Flags: Docs Required
• Assignee: Ravil Galeyev
• Reporter: Alexey Zinoviev
• Created: 16/Aug/18 03:50
• Updated: 31/Oct/19 13:02
• Resolved: 23/Oct/18 16:17

Description

Naive Bayes classifiers are a family of simple probabilistic classifiers based on applying Bayes' theorem with strong (naive) independence assumptions between the features.

So we want to add this algorithm to Apache Ignite ML module.

Ideally, implementation should support both multinomial naive Bayes and Bernoulli naive Bayes.

Requirements for successful PR:

1. PartitionedDataset usage
2. Trainer-Model paradigm support
3. Tests for Model and for Trainer (and other stuff)
4. Example of usage with small, but famous dataset like IRIS, Titanic or House Prices
5. Javadocs/codestyle according guidelines

 

 

Issue Links

• relates to: IGNITE-9978 [ML] Implement Compound Naive Bayes classifier
• links to: Examples
• links to: GitHub Pull Request #4869
• links to: Licenses and JavaDocs
• links to: TC Build