e4a5dbe7e29d64c9b095d134f0ca2a5543353dda,autokeras/hypermodel/node.py,Input,transform,#Input#Any#,28

Before Change

    def transform(self, x):
        Transform x into a compatible type (tf.data.Dataset).
        if isinstance(x, tf.data.Dataset):
            return x
        if isinstance(x, np.ndarray):
            x = x.astype(np.float32)
            return tf.data.Dataset.from_tensor_slices(x)


class ImageInput(Input):
    Input node for image data.

    The input data should be numpy.ndarray or tf.data.Dataset. The shape of the data

After Change

    def transform(self, x):
        Transform x into a compatible type (tf.data.Dataset).
        self._check(x)
        dataset = self._convert_to_dataset(x)
        return dataset


class ImageInput(Input):

Instances