This is the solution for the AI assesment test. The repository contains a utils.py and a jupyter notebook. All the experiments with the evaluation results and plots could be checked in the notebook. Trained models could be find under trained_models/ folder. Plots could be also found in trained_models/plots
Project dependencies could be found in requirements.txt The code was developed on Ubuntu 22.04, using Python 3.10.