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

import ai.h2o.automl.*;
import hex.Model;
import hex.ModelBuilder;
import water.util.Log;


/**
 * This class is decoupled from the XGBoostSteps implementation to avoid having to load XGBoost classes
 * when the extension is not available.
 */
public class XGBoostStepsProvider implements ModelingStepsProvider, ModelParametersProvider {

    @Override
    public String getName() {
        return XGBoostSteps.NAME;
    }

    @Override
    public XGBoostSteps newInstance(AutoML aml) {
        return Algo.XGBoost.enabled() ? new XGBoostSteps(aml) : null;
    }

    @Override
    public Model.Parameters newDefaultParameters() {
        return Algo.XGBoost.enabled() ? ModelBuilder.make(getName(), null, null)._parms : null;
    }
}