Machine Learning for Phase Retrieval from 4D-STEM Data