Can AI help prevent students from dropping out of school?