from sklearn.neighbors import KNeighborsClassifier
from sklearn.decomposition import PCA
import numpy as np
import time

x_tr = np.random.randn(3000, 12*12*31*3)
y_tr = np.random.randint(0, 3, size=(3000, ))

x_te = np.random.randn(10000, 12*12*31*3)

pca = PCA(n_components=100).fit(x_tr)

z_tr = pca.transform(x_tr)
z_te = pca.transform(x_te)

classifier = KNeighborsClassifier(n_neighbors=10, n_jobs=-1).fit(z_tr, y_tr)

time_start = time.time()
predictions = classifier.predict(z_te)
print(time.time() - time_start)