def loadModel(self):
fname = QtGui.QFileDialog.getOpenFileName(self, 'Open Model', '')
self.modelFileNameLabel.setText(fname)
self.pmodel = mpm.read_data_model(fname)
if not ('rates' in self.pmodel.keys()):
self.pmodel['rates'] = [0.0 for _ in range(self.poemCount())]
self.curr_idx = 0
self.poemNumSpin.setMinimum(1)
self.poemNumSpin.setMaximum(self.poemCount())
self.poemNumSpin.setValue(1)
self.nextPoemBtn.setEnabled(True)
self.prevPoemBtn.setEnabled(True)
self.poemNumSpin.setEnabled(True)
self.rateSlider.setEnabled(True)
self.saveModelBtn.setEnabled(True)
def loadModel(self):
fname = QtGui.QFileDialog.getOpenFileName(self, 'Open Model', '')
self.modelFileNameLabel.setText(fname)
self.pmodel = mpm.read_data_model(fname)
if not ('rates' in self.pmodel.keys()):
self.pmodel['rates'] = [0.0 for _ in range(self.poemCount())]
self.curr_idx = 0
self.poemNumSpin.setMinimum(1)
self.poemNumSpin.setMaximum(self.poemCount())
self.poemNumSpin.setValue(1)
self.nextPoemBtn.setEnabled(True)
self.prevPoemBtn.setEnabled(True)
self.poemNumSpin.setEnabled(True)
self.rateSlider.setEnabled(True)
self.saveModelBtn.setEnabled(True)
if __name__ == "__main__":
test_poems = (
"мадам а не хотите кофе\n сказал он глядя ей в глаза\n три таракана были против\n пять за\n",
"в ночи повеяло прохладой\n как будто в дом прокралось зло\n а это просто одеяло сползло\n",
"на юг летят все птицы в мире\n едва лишь осень на порог\n одни пингвины не летают\n пингвинам некуда лететь\n",
"лежу у самой кромки моря\n немножко пьяный ну и пусть\n то море на меня накатит\n то грусть\n",
"олег купил в мясном отделе\n кило можроженных сердец\n и с упоеньем разбивал их\n подлец\n",
"мдам я весь пылаю страстью\n отдайтесь мне среди осин\n я очарован цветом ваших\n лосин\n",
"автобус сделал остановку\n открылась дверь в густой туман\n и я сошла легко и ловко\n с ума",
"когда одна осталась пуля\n а ты врагами окружен\n стреляй последней пулей в солнце\n и уходи пока темно\n",
"горячий чай с пломбиром вместе\n ты ешь смотря цинично вдаль\n скажи дружочек а не треснет\n эмаль",
"на почту бабушки слетелись\n воркуют топчатся голдят\n наверно крошек в виде пенсий\n хотят\n",
"детей кому нибудь подарим\n беспечно отмахнулся пётр\n так был решён вопрос со свадьбой\n с опасным сексом и вобще\n",
"жених идёт и пышет жаром\n и все невесты на селе\n к нему подходят чтоб погреться\n кто посмелее прикурить\n",
"олег оксану в чистом поле\n случайно в полночь повстречал\n и легкомысленно кого-то\n зачал\n",
"зима бежишь с горы кататься\n наперекор своей судьбе\n и кажется что минус двадцать\n тебе\n",
"у истины довольно много\n начал концов и середин\n зато в тупик всегда заходишь\n в один\n",
"чтоб жизнь свою наполнить смыслом\n из вздохов шерсти и тоски\n старушки каждодневно вяжут\n носки\n",
"цветы засушенные в книге\n помада в сумке шоколад\n олег слегка волнуясь прибыл\n в стройбат\n",
"поэт закажет проститутку\n и ночь проводит с ней в стихах\n та иитирует восторги\n ох ах\n",
"маньяк насилует людмилу\n четвёртый час четвёртый раз\n людмила думает вот суки\n про бывших трёх своих мужей\n",
"иван семёныч дерзкий грубый\n брутальный весь такой мужик\n помадой нежно красит губы\n привык\n"
)
w2v_model = sem.load_w2v_model(sem.WORD2VEC_MODEL_FILE)
poems_model = mpm.read_data_model("poems_model.dat")
test_poem = random.choice(test_poems)
sim_poems = similar_poems(test_poem, poems_model, w2v_model)
print(test_poem)
pprint(sim_poems)
if __name__ == "__main__":
test_poems = (
"мадам а не хотите кофе\n сказал он глядя ей в глаза\n три таракана были против\n пять за\n",
"в ночи повеяло прохладой\n как будто в дом прокралось зло\n а это просто одеяло сползло\n",
"на юг летят все птицы в мире\n едва лишь осень на порог\n одни пингвины не летают\n пингвинам некуда лететь\n",
"лежу у самой кромки моря\n немножко пьяный ну и пусть\n то море на меня накатит\n то грусть\n",
"олег купил в мясном отделе\n кило можроженных сердец\n и с упоеньем разбивал их\n подлец\n",
"мдам я весь пылаю страстью\n отдайтесь мне среди осин\n я очарован цветом ваших\n лосин\n",
"автобус сделал остановку\n открылась дверь в густой туман\n и я сошла легко и ловко\n с ума",
"когда одна осталась пуля\n а ты врагами окружен\n стреляй последней пулей в солнце\n и уходи пока темно\n",
"горячий чай с пломбиром вместе\n ты ешь смотря цинично вдаль\n скажи дружочек а не треснет\n эмаль",
"на почту бабушки слетелись\n воркуют топчатся голдят\n наверно крошек в виде пенсий\n хотят\n",
"детей кому нибудь подарим\n беспечно отмахнулся пётр\n так был решён вопрос со свадьбой\n с опасным сексом и вобще\n",
"жених идёт и пышет жаром\n и все невесты на селе\n к нему подходят чтоб погреться\n кто посмелее прикурить\n",
"олег оксану в чистом поле\n случайно в полночь повстречал\n и легкомысленно кого-то\n зачал\n",
"зима бежишь с горы кататься\n наперекор своей судьбе\n и кажется что минус двадцать\n тебе\n",
"у истины довольно много\n начал концов и середин\n зато в тупик всегда заходишь\n в один\n",
"чтоб жизнь свою наполнить смыслом\n из вздохов шерсти и тоски\n старушки каждодневно вяжут\n носки\n",
"цветы засушенные в книге\n помада в сумке шоколад\n олег слегка волнуясь прибыл\n в стройбат\n",
"поэт закажет проститутку\n и ночь проводит с ней в стихах\n та иитирует восторги\n ох ах\n",
"маньяк насилует людмилу\n четвёртый час четвёртый раз\n людмила думает вот суки\n про бывших трёх своих мужей\n",
"иван семёныч дерзкий грубый\n брутальный весь такой мужик\n помадой нежно красит губы\n привык\n"
)
w2v = sem.load_w2v_model(sem.WORD2VEC_MODEL_FILE)
pm = mpm.read_data_model("poems_model.dat")
test_poem = random.choice(test_poems)
sim_poems = similar_poems(test_poem, pm, w2v)
print(test_poem)
pprint(sim_poems)