hex.schemas.GenericModelV3 Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of h2o-algos Show documentation

H2O Algorithms

There is a newer version: 3.46.0.6

package hex.schemas;

import hex.generic.GenericModel;
import hex.generic.GenericModelOutput;
import hex.generic.GenericModelParameters;
import water.api.API;
import water.api.schemas3.ModelOutputSchemaV3;
import water.api.schemas3.ModelSchemaV3;
import water.api.schemas3.TwoDimTableV3;

public class GenericModelV3 extends ModelSchemaV3 {

    @Override
    public GenericV3.GenericParametersV3 createParametersSchema() {
        return new GenericV3.GenericParametersV3();
    }

    @Override
    public GenericModelOutputV3 createOutputSchema() {
        return new GenericModelOutputV3();
    }

    public static final class GenericModelOutputV3 extends ModelOutputSchemaV3{
        @API(help="Variable Importances", direction=API.Direction.OUTPUT, level = API.Level.secondary)
        TwoDimTableV3 variable_importances;
        
        @API(help = "Short identifier of the original algorithm name", direction = API.Direction.OUTPUT,
                level = API.Level.secondary)
        String original_model_identifier;
        
        @API(help = "Full, potentially long name of the original agorithm", direction = API.Direction.OUTPUT,
                level = API.Level.secondary)
        String original_model_full_name;
    }
}