ai.h2o.automl.leaderboard.ModelStep 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 ModelStep extends Iced implements LeaderboardCell {
    public static final LeaderboardColumn COLUMN = new LeaderboardColumn("step", "string", "%s", true);
    final Key _modelId;
    final String _stepId;

    public ModelStep(Model model, ModelingStep step) {
        _modelId = model._key;
        _stepId = step == null ? "" : step.getId();
    }

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

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

    @Override
    public String getValue() {
        return _stepId;
    }

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

}