knn as an input feature