hex.VarImp Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of h2o-core Show documentation

H2O Core

There is a newer version: 3.8.2.9

package hex;

import water.Iced;
import water.util.ArrayUtils;

public class VarImp extends Iced {
  final public float[] _varimp; // Variable importance of individual variables, unscaled
  final public String[] _names; // Names of variables.
  public VarImp(float[] varimp, String[] names) { _varimp = varimp; _names = names; }
  // Scaled, so largest value is 1.0
  public float[] scaled_values() { return ArrayUtils.div (_varimp.clone(),ArrayUtils.maxValue(_varimp)); }
  // Scaled so all elements total to 100%
  public float[] summary()       { return ArrayUtils.mult(_varimp.clone(),100.0f/ArrayUtils.sum(_varimp));
  }
}