プレイリスト生成の検証終了 - 駆け出しエンジニアの作業ノート

コードのtry文を移動させたところ、きちんと動作して最後までたどり着いたので記載させて頂きます。

import psycopg2
import psycopg2.extras
from gensim.models import word2vec

host_name = 
port_number =
dbname = 
rolename = 
passwd = 

def track_search():
    conn = psycopg2.connect(database=dbname, host=host_name, port=port_number, user=rolename, password=passwd)
    cur = conn.cursor()
    cur.execute("SELECT DISTINCT artist, song FROM track_data")
    track_list = cur.fetchall()
    conn.close()
    return track_list

def call_model():
    track_list = track_search()
    high_score = 0
    high_score_track = ""
    search_track = ""
    for i in range(len(track_list)):
        track = track_list[i]
        artist = track[0]
        song = track[1]
        input_data = artist + " - " + song
        model = word2vec.Word2Vec.load("similar_song.model")
        try:
            data = model.most_similar(positive=[input_data])
            high_score_track_data_each_song = data[0]
            high_score_track_each_song = high_score_track_data_each_song[0]
            high_score_each_song = high_score_track_data_each_song[1]
            if high_score_each_song > high_score:
                high_score = high_score_each_song
                high_score_track = high_score_track_each_song
                search_track = track
                print("New record!")
                print(search_track)
                print(high_score_track)
                print(high_score)
            else:
                continue
        except KeyError:
            print("pass")
            continue


if __name__ == '__main__':
    call_model()

結果は以下になり、前日暫定で出した数値を上回りました。

('Teresa Cristina e Grupo Semente', 'A Borboleta e o Passarinho')
KRD8 - 蒼いピストル～らいらら第二章～
0.5622042417526245

これは、「Teresa Cristina e Grupo Semente」の「A Borboleta e o Passarinho」に最も近い曲として「KRD8」の「蒼いピストル～らいらら第二章～」が抽出され類似度の数値は全学習データのうち最も高いという事になります。

O Mundo É Meu Lugar - Ao Vivo

アーティスト: Teresa Cristina
出版社/メーカー: Deckdisc
発売日: 2015/02/27
メディア: MP3 ダウンロード
この商品を含むブログを見る

Dream Rainbow ?次のステージへ一緒に行こう!?

アーティスト: KRD8
出版社/メーカー: UNCLE STAR RECORDS
発売日: 2015/02/04
メディア: CD
この商品を含むブログ (1件) を見る

相変わらず、ジャンルの垣根を越えています…　なお、それ以外の結果は以下の通りです。

[('KRD8 - 蒼いピストル～らいらら第二章～', 0.5622042417526245), ('Jax Jones - Instruction', 0.4197536110877991), ('miwa - サヨナラ', 0.39838463068008423), ('Detlef Keller - Way 5', 0.3839055299758911), ("Ike Quebec - I've Got a Crush On You", 0.38365036249160767), ("Marty Paich - What's new?", 0.35652440786361694), ('a flood of circle - 春の嵐', 0.3534770607948303), ("Gigi Gryce - Nica's Tempo", 0.3506496250629425), ('Martha Argerich - Fantasiestücke, Op 12: Warum?', 0.34342825412750244), ('Wiener Philharmoniker - Smetana: Má Vlast, JB1:112 - 2. Vltava', 0.3430880308151245)]

上記の結果を出したコードは以下を参照して下さい。

psyduck-take-it-easy.hatenablog.com