now_2023.models.CNNModel

class now_2023.models.CNNModel(learning_rate: float, n_epochs: int, batch_size: int)

The CNN model.

__init__(learning_rate: float, n_epochs: int, batch_size: int)

Methods

__init__(learning_rate, n_epochs, batch_size)
fit(img_dir, df[, transform, batch_size])
predict(img_dir, df[, transform, batch_size])
save(output_filename)