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

import ai.h2o.automl.AutoML;
import water.Iced;

public class PreprocessingStepDefinition extends Iced {
    
    public enum Type {
        TargetEncoding
    }

    Type _type;

    public PreprocessingStepDefinition() { /* for reflection */ }

    public PreprocessingStepDefinition(Type type) { 
        _type = type;
    }

    public PreprocessingStep newPreprocessingStep(AutoML aml) {
        switch (_type) {
            case TargetEncoding:
                return new TargetEncoding(aml);
            default:
                throw new IllegalStateException();
        }
    }
}