Yes, you should be able to transform the stored features to numpy arrays (if they are stored as tensors, use arr = tensor.numpy()) and train a KNN classifier via e.g. sklearn.neighbors.KNeighborsClassifier 4.