Package org.apache.ignite.examples.ml.preprocessing.encoding

Class TargetEncoderExample

  • public class TargetEncoderExample
extends java.lang.Object
    Example that shows how to use Target Encoder preprocessor to encode labels presented as a mean target value.

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

    After that it defines preprocessors that extract features from an upstream data and encode category with avarage target value (categories).

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

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

    Daniele Miccii-Barreca (2001). A Preprocessing Scheme for High-Cardinality Categorical Attributes in Classification and Prediction Problems. SIGKDD Explor. Newsl. 3, 1. From http://dx.doi.org/10.1145/507533.507538

    • 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

      • TargetEncoderExample

        public TargetEncoderExample()

    • Method Detail

      • main

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