hex.genmodel.attributes.SharedTreeModelAttributes 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.attributes;

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


public class SharedTreeModelAttributes extends ModelAttributes implements VariableImportancesHolder {

  private final VariableImportances _variableImportances;

  public SharedTreeModelAttributes(JsonObject modelJson, MojoModel model) {
    super(model, modelJson);
    _variableImportances = VariableImportances.extractFromJson(modelJson);

  }

  /**
   * @return A {@link VariableImportances} instance with variable importances for each feature used.
   */
  public VariableImportances getVariableImportances() {
    return _variableImportances;
  }
  
}