cdbb8791ec13d03fd1fcf2580111db677a3ebc29,autokeras/hypermodel/block.py,EmbeddingBlock,set_state,#EmbeddingBlock#Any#,564

Before Change

        return state

    def set_state(self, state):
        super().set_state(state)
        self.max_features = state["max_features"]
        self.pretraining = state["pretraining"]
        self.embedding_dim = state["embedding_dim"]

After Change

    def build(self, hp, inputs=None):
        inputs = nest.flatten(inputs)
        if len(inputs) == 1:
            return inputs

        merge_type = self.merge_type or hp.Choice("merge_type",
                                                  ["add", "concatenate"],
                                                  default="add")

Instances