taxi_data_matrix = csr_matrix(taxi_data_pivot.values) from sklearn.neighbors import NearestNeighbors model_knn = NearestNeighbors(metric = 'cosine', algorithm='brute') model_knn.fit(taxi_data_matrix)