Examples with GLMModel hex.glm.GLMModel used on opensource projects

Example 11 with GLMModel

use of hex.glm.GLMModel in project h2o-3 by h2oai.

the class ModelSerializationTest method prepareGLMModel.

private GLMModel prepareGLMModel(String dataset, String[] ignoredColumns, String response, GLMModel.GLMParameters.Family family) {
    Frame f = parse_test_file(dataset);
    try {
        GLMModel.GLMParameters params = new GLMModel.GLMParameters();
        params._train = f._key;
        params._ignored_columns = ignoredColumns;
        params._response_column = response;
        params._family = family;
        return new GLM(params).trainModel().get();
    } finally {
        if (f != null)
            f.delete();
    }
}