Adapt this implementation to other data

[nk-fouque]

Hi, I understand that this was made to replicate an example, but I would like to use it for may classification task, What part of code do I have to change to train it on my data ?