Source code for autoflow.pipeline.components.preprocessing.impute.fill_abnormal

from typing import Dict

from autoflow.pipeline.components.feature_engineer_base import AutoFlowFeatureEngineerAlgorithm

__all__ = ["FillAbnormal"]


[docs]class FillAbnormal(AutoFlowFeatureEngineerAlgorithm):
    class__ = "BaseImputer"
    module__ = "sklearn.impute"

[docs]    def after_process_hyperparams(self, hyperparams) -> Dict:
        hyperparams = super(FillAbnormal, self).after_process_hyperparams(hyperparams)
        hyperparams["fill_value"] = -999
        hyperparams["strategy"] = "constant"
        return hyperparams