Package org.apache.ignite.examples.ml.selection.split

Class TrainTestDatasetSplitterExample

  • public class TrainTestDatasetSplitterExample
extends java.lang.Object
    Run linear regression model over dataset split on train and test subsets (TrainTestDatasetSplitter).

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

    After that it creates dataset splitter and trains the linear regression model based on the specified data using this splitter.

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

    You can change the test data and split parameters used in this example and re-run it to explore this functionality 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

      • TrainTestDatasetSplitterExample

        public TrainTestDatasetSplitterExample()

    • Method Detail

      • main

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