ai.h2o.automl.leaderboard.ModelGroup Maven / Gradle / Ivy

Go to download

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

H2O AutoML

The newest version!

package ai.h2o.automl.leaderboard;

import ai.h2o.automl.ModelingStep;
import hex.Model;
import hex.leaderboard.LeaderboardCell;
import hex.leaderboard.LeaderboardColumn;
import water.Iced;
import water.Key;

public class ModelGroup extends Iced implements LeaderboardCell {
    public static final LeaderboardColumn COLUMN = new LeaderboardColumn("group", "int", "%s", true);
    final Key _modelId;
    final int _priorityGroup;

    public ModelGroup(Model model, ModelingStep step) {
        _modelId = model._key;
        _priorityGroup = step == null ? -1 : step.getPriorityGroup();
    }

    @Override
    public LeaderboardColumn getColumn() {
        return COLUMN;
    }

    @Override
    public Key getModelId() {
        return _modelId;
    }

    @Override
    public Integer getValue() {
        return _priorityGroup;
    }

    @Override
    public void setValue(Integer value) {
        throw new UnsupportedOperationException();
    }

}