tflite_model_maker.question_answer.create

Stay organized with collections Save and categorize content based on your preferences.

Loads data and train the model for question answer.

@classmethod
tflite_model_maker.question_answer.create(
    train_data,
    model_spec,
    batch_size=None,
    epochs=2,
    steps_per_epoch=None,
    shuffle=False,
    do_train=True
)

Used in the notebooks

Used in the tutorials

train_data Training data.
model_spec Specification for the model.
batch_size Batch size for training.
epochs Number of epochs for training.
steps_per_epoch Integer or None. Total number of steps (batches of samples) before declaring one epoch finished and starting the next epoch. If steps_per_epoch is None, the epoch will run until the input dataset is exhausted.
shuffle Whether the data should be shuffled.
do_train Whether to run training.

An instance based on QuestionAnswer.