Neuroglycerin / neukrill-net-tools

Tools coded as part of the NDSB competition.
MIT License
0 stars 0 forks source link

Taxonomy integration #123

Closed gngdb closed 9 years ago

gngdb commented 9 years ago

We want to be able to train a hierarchical model that'll assign probabilities to superclasses and sub-classes with a whole set of Pylearn2 models. To do this, we need taxonomy.py to be integrated with the main codebase.

We need to be able to pass an option in run settings to transform the classes and image_fnames dictionary stored in the settings structure, and we have to make sure this is done in the dataset object. Unfortunately, settings is loaded before run settings. You'll have to decide how to deal with that and mention it here.

The transformation is fairly simple, but there are two types of transformation we want to be able to do:

gngdb commented 9 years ago

Once this is done we're most of the way towards being able to train the models that make up the model described in this issue.

gngdb commented 9 years ago

Training of this model was succesful, was able to reach similar validation set levels of performance, but then overfit and only improved on the test set. Details should be in a notebook called Hierarchical Run.