clf = KMeans(n_clusters=10)

s = time.time()
clf.fit(x_train)
e = time.time()
print("Training time = {}".format(e - s))


s = time.time()
clf.predict(x_test)
e = time.time()
print("Prediction time = {}".format((e - s) / len(y_test)))