Transform a name or description into a color representation
This is the code used in this article where the paper Character Sequence Models for Colorful Words is analyzed.
This code was tested on Python 3.6 / Pytorch 1.0
-
Modify
settings.py
according to your system. -
run_download_data.py
will download tuples (color, description) from ColourLovers using their API. -
run_process_data.py
will clean and split the data. -
run_train.py
will train and save the best model. -
run_inference.py
andrun_incremental_inference.py
are used to obtain the reuslts on the test set.
Most of the content on this repository is based on the great Pytorch tutorials by epochx