Package org.apache.ignite.examples.ml.naivebayes

Class CompoundNaiveBayesExample

  • public class CompoundNaiveBayesExample
extends java.lang.Object
    Run naive Compound Bayes classification model based on Nnaive Bayes classifier algorithm (GaussianNaiveBayesTrainer)and Discrete naive Bayes classifier algorithm (DiscreteNaiveBayesTrainer) over distributed cache.

    Code in this example launches Ignite grid and fills the cache with test data points.

    After that it trains the naive Bayes classification model based on the specified data.

    Finally, this example loops over the test set of data points, applies the trained model to predict the target value, compares prediction to expected outcome (ground truth), and builds confusion matrix.

    You can change the test data used in this example and re-run it to explore this algorithm further.

    • Method Summary

      All Methods Static Methods Concrete Methods 
      Modifier and Type Method Description
      static void main​(java.lang.String[] args)
      Run example.

      • Methods inherited from class java.lang.Object

        clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

    • Constructor Detail

      • CompoundNaiveBayesExample

        public CompoundNaiveBayesExample()

    • Method Detail

      • main

        public static void main​(java.lang.String[] args)
                 throws java.io.IOException
        Run example.
        Throws:
        java.io.IOException