Concept Classifier Model

Getting started

Install dependencies:

pip install -r requirements.txt

Hyperparameters study

The script hyperparameters.py searches a given space of hyperparameters to find the best combination. It takes a while to finish and it is best to run on GPUs. It can be run as follows:

python scripts/hyperparameters.py

Notebook

Once the optimal hyperparameters have been found, model can be trained using the notebook entrainement_modeles_concepts.ipynb.

Model

The working model delivered by Joffrey can be found in the drive