Package org.apache.ignite.examples.ml.tutorial

Class Step_7_Split_train_test

  • public class Step_7_Split_train_test
extends java.lang.Object
    The highest accuracy in the previous example (Step_6_KNN) is the result of overfitting. For real model estimation is better to use test-train split via TrainTestDatasetSplitter.

    Code in this example launches Ignite grid and fills the cache with test data (based on Titanic passengers data).

    After that it defines how to split the data to train and test sets and configures preprocessors that extract features from an upstream data and perform other desired changes over the extracted data.

    Then, it trains the model based on the processed data using decision tree classification.

    Finally, this example uses Evaluator functionality to compute metrics from predictions.

    • 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

      • Step_7_Split_train_test

        public Step_7_Split_train_test()

    • Method Detail

      • main

        public static void main​(java.lang.String[] args)
        Run example.