Automate data categorization with AI