hex.genmodel.attributes.DeepLearningModelAttributes 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 DeepLearningModelAttributes extends ModelAttributes implements VariableImportancesHolder {

    private final VariableImportances _variableImportances;

    public DeepLearningModelAttributes(MojoModel model, JsonObject modelJson) {
        super(model, modelJson);
        _variableImportances = VariableImportances.extractFromJson(modelJson);
    }
    
    public VariableImportances getVariableImportances(){
        return _variableImportances;
    }
}