extend and expose KMeansDataGenerator