AI for data analysis