Node

[source]

ImageInput

autokeras.ImageInput(name=None, **kwargs)

Input node for image data.

The input data should be numpy.ndarray or tf.data.Dataset. The shape of the data should be should be (samples, width, height) or (samples, width, height, channels).

Arguments

• name str | None: String. The name of the input node. If unspecified, it will be set automatically with the class name.

---

[source]

Input

autokeras.Input(name=None, **kwargs)

Input node for tensor data.

The data should be numpy.ndarray or tf.data.Dataset.

Arguments

• name str | None: String. The name of the input node. If unspecified, it will be set automatically with the class name.

---

[source]

TextInput

autokeras.TextInput(name=None, **kwargs)

Input node for text data.

The input data should be numpy.ndarray or tf.data.Dataset. The data should be one-dimensional. Each element in the data should be a string which is a full sentence.

Arguments

• name str | None: String. The name of the input node. If unspecified, it will be set automatically with the class name.

---