Thanks for you amazing work! Can I have the train/fit and predict code of a case study? Thanks again.