Using AI to help understand the evolution of young stars and their planets