cdbb8791ec13d03fd1fcf2580111db677a3ebc29,autokeras/hypermodel/preprocessor.py,FeatureEngineering,get_config,#FeatureEngineering#,810

Before Change

            weights["high_level_num_cat"], ast.literal_eval)

    def get_config(self):
        return {"column_names": self.column_names,
                "column_types": utils.to_type_key(self.column_types, str),
                "num_columns": self.num_columns,
                "max_columns": self.max_columns}

    def set_config(self, config):
        self.column_names = config["column_names"]
        self.column_types = config["column_types"]

After Change

            state["high_level_num_cat"], ast.literal_eval)

    def get_config(self):
        config = super().get_config()
        config.update({"max_columns": self.max_columns})
        return config

Instances