You have a requirement to train and apply different machine learning models to different groups within a dataset. Which Pandas Function API would you use to achieve this?