import json
import os


# define dataset
# using training from https://www.kaggle.com/slothkong/10-monkey-species
# TODO load dataset


# TODO train/validation split (or something else...)

# define model
def define_model(input_shape, output_class_number):
    model = None
    # TODO define the network architecture

    return model

# TODO define your training loop + performance indicators (tips: ignite helps)


# TODO instantiate your model


# TODO run it


# TODO don't forget to save the model and weights

