script for KNN regression