ai.h2o.automl.ModelSelectionStrategy 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;

import hex.Model;
import water.Key;

@FunctionalInterface
public interface ModelSelectionStrategy{

    class Selection {
        final Key[] _add;  //models that should be added to the original population
        final Key[] _remove; //models that should be removed from the original population

        public Selection(Key[] add, Key[] remove) {
            _add = add;
            _remove = remove;
        }
    }

    Selection select(Key[] originalModels, Key[] newModels);
}