I'm shocked by the work you've done. But could you provide the model training code for me to understand?