code for down size the test set for word embed algo.