hex.genmodel.CategoricalEncoding Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of h2o-genmodel Show documentation
H2O GenModel
There is a newer version: 3.46.0.5
package hex.genmodel;

import hex.genmodel.easy.*;

import java.util.Map;

public enum CategoricalEncoding {
  AUTO(false) {
    @Override
    public Map createColumnMapping(GenModel m) {
      return new EnumEncoderColumnMapper(m).create();
    }

    @Override
    public Map createCategoricalEncoders(GenModel m, Map columnMapping) {
      return new EnumEncoderDomainMapConstructor(m, columnMapping).create();
    }
  },
  OneHotExplicit(false) {
    @Override
    public Map createColumnMapping(GenModel m) {
      return new OneHotEncoderColumnMapper(m).create();
    }

    @Override
    public Map createCategoricalEncoders(GenModel m, Map columnMapping) {
      return new OneHotEncoderDomainMapConstructor(m, columnMapping).create();
    }
  },
  Binary(false) {
    @Override
    public Map createColumnMapping(GenModel m) {
      return new BinaryColumnMapper(m).create();
    }

    @Override
    public Map createCategoricalEncoders(GenModel m, Map columnMapping) {
      return new BinaryDomainMapConstructor(m, columnMapping).create();
    }
  },
  EnumLimited(true) {
    @Override
    public Map createColumnMapping(GenModel m) {
      return new EnumLimitedEncoderColumnMapper(m).create();
    }

    @Override
    public Map createCategoricalEncoders(GenModel m, Map columnMapping) {
      return new EnumLimitedEncoderDomainMapConstructor(m, columnMapping).create();
    }
  },
  Eigen(true) {
    @Override
    public Map createColumnMapping(GenModel m) {
      return new EigenEncoderColumnMapper(m).create();
    }

    @Override
    public Map createCategoricalEncoders(GenModel m, Map columnMapping) {
      return new EigenEncoderDomainMapConstructor(m, columnMapping).create();
    }
  },
  LabelEncoder(false) {
    @Override
    public Map createColumnMapping(GenModel m) {
      return new EnumEncoderColumnMapper(m).create();
    }

    @Override
    public Map createCategoricalEncoders(GenModel m, Map columnMapping) {
      return new LabelEncoderDomainMapConstructor(m, columnMapping).create();
    }
  };

  private final boolean _parametrized;

  CategoricalEncoding(boolean parametrized) {
    _parametrized = parametrized;
  }

  public abstract Map createColumnMapping(GenModel m);

  public abstract Map createCategoricalEncoders(GenModel m, Map columnMapping);

  /**
   * Does the categorical encoding have any parameters that are needed to correctly interpret it?
   * Eg.: number of classes for EnumLimited
   * 
   * @return Is this encoding parametrized?
   */
  public boolean isParametrized() {
    return _parametrized;
  }

}