Data analysis for ai models