Structure prediction in high-entropy alloys with machine learning