public class TargetEncoding extends java.lang.Object implements PreprocessingStep
PreprocessingStep.Completer| Modifier and Type | Field and Description |
|---|---|
static java.lang.String |
CONFIG_ENABLED |
static java.lang.String |
CONFIG_PREPARE_CV_ONLY |
| Constructor and Description |
|---|
TargetEncoding(AutoML aml) |
| Modifier and Type | Method and Description |
|---|---|
PreprocessingStep.Completer |
apply(hex.Model.Parameters params,
PreprocessingConfig config)
applies this preprocessing step to the model parameters right before the model training starts.
|
static water.fvec.Vec |
createFoldColumn(water.fvec.Frame fr,
hex.Model.Parameters.FoldAssignmentScheme fold_assignment,
int nfolds,
java.lang.String responseColumn,
long seed) |
void |
dispose()
preprocessing steps are disposed by default at the end of the AutoML training session.
|
java.lang.String |
getType() |
void |
prepare()
preprocessing steps are prepared by default before the AutoML session starts training the first model.
|
void |
remove()
Completely remove from the system
|
void |
setColumnCardinalityThreshold(int threshold) |
void |
setDefaultParams(ai.h2o.targetencoding.TargetEncoderModel.TargetEncoderParameters defaultParams) |
void |
setEncodeAllColumns(boolean encodeAllColumns) |
public static java.lang.String CONFIG_ENABLED
public static java.lang.String CONFIG_PREPARE_CV_ONLY
public TargetEncoding(AutoML aml)
public java.lang.String getType()
getType in interface PreprocessingSteppublic void prepare()
PreprocessingStepprepare in interface PreprocessingSteppublic PreprocessingStep.Completer apply(hex.Model.Parameters params, PreprocessingConfig config)
PreprocessingStepapply in interface PreprocessingSteppublic void dispose()
PreprocessingStepdispose in interface PreprocessingSteppublic void remove()
PreprocessingStepremove in interface PreprocessingSteppublic void setDefaultParams(ai.h2o.targetencoding.TargetEncoderModel.TargetEncoderParameters defaultParams)
public void setEncodeAllColumns(boolean encodeAllColumns)
public void setColumnCardinalityThreshold(int threshold)
public static water.fvec.Vec createFoldColumn(water.fvec.Frame fr,
hex.Model.Parameters.FoldAssignmentScheme fold_assignment,
int nfolds,
java.lang.String responseColumn,
long seed)