add tests and Word2VecModelWrapper
cache words RDD in fit