学習済みモデル

学習済みにおける各単語の重みを表示します

from sklearn.externals import joblib
f = open('result/weight.txt', 'a')
if __name__ == '__main__':
    model = joblib.load('result/model.pkl')
    vocab = joblib.load('result/vocab.pkl')

    # 学習済モデルの確認
    for weight, word in sorted(zip(model.coef_[0], vocab), reverse=True):
        f.write(str('{0:f} {1:s}'.format(weight, word)) + '\n')
        print('{0:f} {1:s}'.format(weight, word))
f.close()

結果

0.591078 バラ
0.584908 示す
0.566019 占領
0.557807 調理
0.551214 完全
0.547697 恒常
0.543960 非難
0.529786 森林地帯
0.526189 座礁
0.525207 地区
0.510096 ブッシュ
0.501339 続ける
0.494787 ステップ
0.486528 買上げ
0.486528 ほか
0.483724 ニシン
0.483031 年間
0.476800 ヒツジ
・・・
-0.207971 穀粉
-0.208848 決定
-0.211129 アレンジ
-0.211299 ゴールド
-0.211699 衣料
-0.211699 切れ目
-0.211699 一方
-0.212159 源流
-0.212452 地下
-0.213936 もたらす
-0.214086 原油