Models

The model module.

This module contains the machine learning and deep learning models used in the tutorial.

Classes:

ToyModel(X, y)

Toy model for classification.

CNNModel(learning_rate, n_epochs, batch_size)

The CNN model.