train a separate model for each category :1234: