e709576eb39ac3ddb1f59d291178b7364bdf068e,autokeras/preprocessors/common.py,CategoricalToNumerical,get_config,#CategoricalToNumerical#,137

Before Change

            "column_types": self.column_types,
            "column_names": self.column_names,
            "layer": preprocessors.serialize(self.layer),
            "encoding_layers": preprocessors.serialize(self.layer.encoding_layers),
        }

    @classmethod

After Change

    def get_config(self):
        vocab = []
        for encoding_layer in self.encoding_layers:
            if encoding_layer is None:
                vocab.append([])
            else:
                vocab.append(encoding_layer.get_vocabulary())
        return {
            "column_types": self.column_types,
            "column_names": self.column_names,
            "layer": preprocessors.serialize(self.layer),

Instances