hex.genmodel.algos.kmeans.KMeansMojoModel 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.algos.kmeans;

import hex.genmodel.IClusteringModel;
import hex.genmodel.MojoModel;

public class KMeansMojoModel extends MojoModel implements IClusteringModel {

  boolean _standardize;
  double[][] _centers;

  double[] _means;
  double[] _mults;
  int[] _modes;

  KMeansMojoModel(String[] columns, String[][] domains, String responseColumn) {
    super(columns, domains, responseColumn);
  }

  @Override
  public double[] score0(double[] row, double[] preds) {
    if (_standardize)
      Kmeans_preprocessData(row, _means, _mults, _modes);
    preds[0] = KMeans_closest(_centers, row, _domains);
    return preds;
  }

  @Override
  public int distances(double[] row, double[] distances) {
    if (_standardize)
      Kmeans_preprocessData(row, _means, _mults, _modes);
    return KMeans_distances(_centers, row, _domains, distances);
  }

  @Override
  public int getNumClusters() {
    return _centers.length;
  }

}