Package org.apache.ignite.examples.ml.tree.randomforest

Class RandomForestClassificationExample

  • public class RandomForestClassificationExample
extends java.lang.Object
    Example represents a solution for the task of wine classification based on a Random Forest implementation for multi-classification.

    Code in this example launches Ignite grid and fills the cache with test data points (based on the Wine recognition dataset).

    After that it initializes the RandomForestClassifierTrainer with thread pool for multi-thread learning and trains the model based on the specified data using random forest regression algorithm.

    Finally, this example loops over the test set of data points, compares prediction of the trained model to the expected outcome (ground truth), and evaluates accuracy of the model.

    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

      • RandomForestClassificationExample

        public RandomForestClassificationExample()

    • Method Detail

      • main

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