Machine Learning for K-adaptability in Two-stage Robust Optimization