Is there a way to extract the K nearest neighbors from Training samples from the KNN model in the Scala version?