knn_strength.py: script to determine the MSE of a strength test set