sklearn-style interface to MLLib KMeans clustering