experimented with weights in kmeans.py

for dist_cos metric

added function to zero out weights in a distribution