a832ad85e57224dfa57e51ae75fb45faf8ac0ca6,docs/py/text_classification.py,,,#,17

Before Change

import tensorflow as tf
train_set = tf.data.Dataset.from_tensor_slices(((x_train, ), (y_train, )))
test_set = tf.data.Dataset.from_tensor_slices(((x_test, ), (y_test, )))

clf = ak.TextClassifier(max_trials=10)
// Feed the tensorflow Dataset to the classifier.

After Change

import tensorflow as tf
train_set = tf.data.Dataset.from_tensor_slices(((x_train, ), (y_train, ))).batch(32)
test_set = tf.data.Dataset.from_tensor_slices(((x_test, ), (y_test, ))).batch(32)

clf = ak.TextClassifier(max_trials=3)
// Feed the tensorflow Dataset to the classifier.

Instances