Word To Vec Python