Update gensim_test.py

Tried gensim at the beginning, but I decided to use scikit-learn library later