Attention: This MediaPipe Solutions Preview is an early release. Learn more.

mediapipe_model_maker.image_classifier.train_image_classifier_lib.train_model

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

Trains model with the input data and hyperparameters.

model Input tf.keras.Model.
hparams Hyperparameters for training image classifier.
train_ds tf.data.Dataset, training data to be fed in tf.keras.Model.fit().
validation_ds tf.data.Dataset, validation data to be fed in tf.keras.Model.fit().

The tf.keras.callbacks.History object returned by tf.keras.Model.fit().