hex.genmodel.attributes.ModelAttributesGLM 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.6

package hex.genmodel.attributes;

import com.google.gson.JsonObject;
import hex.genmodel.MojoModel;
import hex.genmodel.attributes.parameters.VariableImportancesHolder;

public class ModelAttributesGLM extends ModelAttributes implements VariableImportancesHolder {

  public final Table _coefficients_table;
  private final VariableImportances _variableImportances;

  public ModelAttributesGLM(MojoModel model, JsonObject modelJson) {
    super(model, modelJson);
    _coefficients_table = ModelJsonReader.readTable(modelJson, "output.coefficients_table");
    _variableImportances = VariableImportances.extractFromJson(modelJson);
  }

  public VariableImportances getVariableImportances(){
    return _variableImportances;
  }

}