hex.schemas.GLMModelV3 Maven / Gradle / Ivy

Go to download
package hex.schemas;

import hex.glm.GLMModel;
import water.api.API;
import water.api.ModelOutputSchema;
import water.api.ModelSchema;

import water.util.ArrayUtils;
import water.api.TwoDimTableBase;
import water.util.TwoDimTable;

import java.util.Arrays;
import java.util.Comparator;
//import water.util.DocGen.HTML;

public class GLMModelV3 extends ModelSchema {

  public static final class GLMModelOutputV3 extends ModelOutputSchema {

    @API(help="Table of Coefficients")
    TwoDimTableBase coefficients_table;

    @API(help="Standardized Coefficient Magnitudes")
    TwoDimTableBase standardized_coefficients_magnitude;


    @Override
    public GLMModelOutputV3 fillFromImpl(GLMModel.GLMOutput impl) {
      super.fillFromImpl(impl);
      String [] names = impl.coefficientNames().clone();
      // put intercept as the first
      String [] ns = ArrayUtils.append(new String[]{"Intercept"},Arrays.copyOf(names,names.length-1));
      coefficients_table = new TwoDimTableBase();
      final double [] magnitudes;
      double [] beta = impl.beta();
      if(impl.isStandardized()){
        double [] norm_beta = impl.getNormBeta();
        // coefficients_table = new TwoDimTable("Coefficients",impl._names,impl.isNormalized()? new String[]{"Coefficients, Normalized Coefficients"}: new String[]{"Coefficients"});
        String [] colTypes = new String[]{"double","double"};
        String [] colFormats = new String[]{"%5f", "%5f"};
        TwoDimTable tdt = new TwoDimTable("Coefficients","glm coefficients", ns, new String[]{"Coefficients", "Standardized Coefficients"}, colTypes, colFormats, "names");
        tdt.set(0,0,beta[beta.length-1]);
        tdt.set(0,1,norm_beta[norm_beta.length-1]);
        for(int i = 0; i < beta.length-1; ++i) {
          tdt.set(i+1, 0, beta[i]);
          tdt.set(i+1, 1, norm_beta[i]);
        }
        coefficients_table.fillFromImpl(tdt);
        magnitudes = norm_beta.clone();
        for(int i = 0; i < magnitudes.length; ++i)
          if(magnitudes[i] < 0) magnitudes[i] *= -1;

        Integer [] indices = new Integer[magnitudes.length-1];
        for(int i = 0; i < indices.length; ++i)
          indices[i] = i;

        Arrays.sort(indices, new Comparator() {
          @Override
          public int compare(Integer o1, Integer o2) {
            if(magnitudes[o1] < magnitudes[o2]) return +1;
            if(magnitudes[o1] > magnitudes[o2]) return -1;
            return 0;
          }
        });
        String [] names2 = new String[names.length];
        for(int i = 0; i < names2.length-1; ++i)
          names2[i] = names[indices[i]];
        tdt = new TwoDimTable("Standardized Coefficient Magnitudes","standardized coefficient magnitudes", names2, new String[]{"Coefficients", "Sign"},new String[]{"double","string"},new String[]{"%5f","%s"},"names");
        for(int i = 0; i < beta.length-1; ++i) {
          tdt.set(i, 0, magnitudes[indices[i]]);
          tdt.set(i, 1, beta[indices[i]] < 0?"NEG":"POS");
        }
        standardized_coefficients_magnitude = new TwoDimTableBase();
        standardized_coefficients_magnitude.fillFromImpl(tdt);
      } else {
        // coefficients_table = new TwoDimTable("Coefficients",impl._names,impl.isNormalized()? new String[]{"Coefficients, Normalized Coefficients"}: new String[]{"Coefficients"});
        String [] colTypes = new String[]{"double"};
        String [] colFormats = new String[]{"%5f"};
        TwoDimTable tdt = new TwoDimTable("Coefficients","glm coefficients", ns, new String[]{"Coefficients"}, colTypes, colFormats, "names");
        tdt.set(0,0,beta[beta.length-1]);
        for(int i = 0; i < beta.length-1; ++i) {
          tdt.set(i + 1, 0, beta[i]);
        }
        coefficients_table.fillFromImpl(tdt);
        magnitudes = beta.clone();
        for(int i = 0; i < magnitudes.length-1; ++i)
          if(magnitudes[i] < 0) magnitudes[i] *= -1;
      }
      return this;
    }
  } // GLMModelOutputV2

  public GLMV3.GLMParametersV3 createParametersSchema() { return new GLMV3.GLMParametersV3(); }
  public GLMModelOutputV3 createOutputSchema() { return new GLMModelOutputV3(); }

  @Override public GLMModel createImpl() {
    GLMModel.GLMParameters parms = parameters.createImpl();
    return new GLMModel( model_id.key(), parms, null, 0.0, 0.0, 0.0, 0, false, false);
  }
}