Have a question? Connect with the community at the TensorFlow Forum Visit Forum

tflite_model_maker.text_classifier.create

Loads data and train the model for test classification.

Used in the notebooks

Used in the tutorials

train_data Training data.
model_spec Specification for the model.
validation_data Validation data. If None, skips validation process.
batch_size Batch size for training.
epochs Number of epochs for training.
shuffle Whether the data should be shuffled.
do_train Whether to run training.

An instance based on TextClassifier.