Training the drawing classification model