hex.util.ClusteringUtils Maven / Gradle / Ivy

Go to download

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

H2O Algorithms

There is a newer version: 3.46.0.6

package hex.util;

import hex.ClusteringModel;
import water.util.TwoDimTable;

public class ClusteringUtils {

    static public TwoDimTable createCenterTable(ClusteringModel.ClusteringOutput output, boolean standardized) {
        String name = standardized ? "Standardized Cluster Means" : "Cluster Means";
        if(output._size == null || output._names == null || output._domains == null || output._centers_raw == null ||
                (standardized && output._centers_std_raw == null)) {
            TwoDimTable table = new TwoDimTable(name, null, new String[] {"1"}, new String[]{"C1"}, new String[]{"double"},
                    new String[]{"%f"}, "Centroid");
            table.set(0,0,Double.NaN);
            return table;
        }

        String[] rowHeaders = new String[output._size.length];
        for(int i = 0; i < rowHeaders.length; i++)
            rowHeaders[i] = String.valueOf(i+1);
        String[] colTypes = new String[output._names.length];
        String[] colFormats = new String[output._names.length];
        for (int i = 0; i < output._domains.length; ++i) {
            colTypes[i] = output._domains[i] == null ? "double" : "String";
            colFormats[i] = output._domains[i] == null ? "%f" : "%s";
        }
        TwoDimTable table = new TwoDimTable(name, null, rowHeaders, output._names, colTypes, colFormats, "Centroid");

        // Internal weights/folds column is included in domain length
        int domain_length = output.hasWeights()? output._domains.length - 1 : output._domains.length;
        for (int j=0; j < domain_length; ++j) {
            boolean string = output._domains[j] != null;
            if (string) {
                for (int i=0; i