Explore ML training for the new dataset with a new temperature dimension