41c99fbf385a8c875fb6181ce7301e4bc218535b,autokeras/keras_layers.py,CategoricalEncoding,call,#CategoricalEncoding#Any#,58

Before Change

            else:
                col = tf.strings.to_number(col, tf.float32)
            outputs.append(col)
        outputs = tf.concat(outputs, axis=-1)
        outputs.set_shape(inputs.shape)
        return outputs

    def compute_output_shape(self, input_shape):
        return input_shape

After Change

        input_nodes = nest.flatten(inputs)[0]
        split_inputs = tf.split(input_nodes, [1] * len(self.encoding), axis=-1)
        output_nodes = []
        for input_node, encoding_layer in zip(split_inputs, self.encoding_layers):
            if encoding_layer is None:
                output_nodes.append(tf.strings.to_number(input_node, tf.float32))
            else:
                output_nodes.append(tf.cast(encoding_layer(input_node), tf.float32))
        return tf.keras.layers.Concatenate()(output_nodes)

    def adapt(self, data):
        for index, encoding_layer in enumerate(self.encoding_layers):

Instances