hex.ensemble.XGBoostMetalearnerProvider Maven / Gradle / Ivy

Go to download

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

H2O XGBoost support

There is a newer version: 3.46.0.6

package hex.ensemble;

import hex.genmodel.utils.DistributionFamily;
import hex.schemas.XGBoostV3;

public class XGBoostMetalearnerProvider implements MetalearnerProvider {

    static class XGBoostMetalearner extends Metalearners.MetalearnerWithDistribution {
        public XGBoostMetalearner() {
            super(Algorithm.xgboost.name());
        }
    }

    @Override
    public String getName() {
        return Metalearner.Algorithm.xgboost.name();
    }

    @Override
    public XGBoostMetalearner newInstance() {
        return new XGBoostMetalearner();
    }

    @Override
    public XGBoostV3.XGBoostParametersV3 newParametersSchemaInstance() {
        return new XGBoostV3.XGBoostParametersV3();
    }
}