Uploaded image for project: 'Ignite'
  1. Ignite
  2. IGNITE-9281 [ML] Starter ML tasks
  3. IGNITE-9747

[ML] Add Bernoulli Naive Bayes classifier

    XMLWordPrintableJSON

    Details

    • Type: Sub-task
    • Status: Resolved
    • Priority: Major
    • Resolution: Done
    • Affects Version/s: None
    • Fix Version/s: 2.8
    • Component/s: ml
    • Labels:
    • Ignite Flags:
      Docs Required

      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.

      Bernoulli Naive Bayes implements the naive Bayes training and classification algorithms for data that is distributed according to multivariate Bernoulli distributions; i.e., there may be multiple features but each one is assumed to be a binary-valued (Bernoulli, boolean) variable.

      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

        Attachments

          Issue Links

            Activity

              People

              • Assignee:
                rgaleyev Ravil Galeyev
                Reporter:
                rgaleyev Ravil Galeyev
              • Votes:
                1 Vote for this issue
                Watchers:
                4 Start watching this issue

                Dates

                • Created:
                  Updated:
                  Resolved: