Add a script to generate the classification model