Enabling antenna design with nano-magnetic materials using machine learning