Documentation
Modeling
Classification - Images
Train and Evaluate

Module: train_and_evaluate.py

Functions

def train_and_evaluate(model, train_generator, validation_generator, epochs: int):
    """
    Train and evaluate the image classification model.
 
    Parameters:
        model (tf.keras.Model): The compiled Keras model instance.
        train_generator (tf.keras.preprocessing.image.DirectoryIterator): Data generator for training data.
        validation_generator (tf.keras.preprocessing.image.DirectoryIterator): Data generator for validation data.
        epochs (int): Number of epochs to train.
 
    Returns:
        tf.keras.callbacks.History: Training history for further analysis.
    """