pytorch sentiment neuron
1.0.0
Pytorch、Cuda、Python 3.5が必要です
Pytorch version of Generating-reviews-Discovering-sentiment:https://github.com/openai/generating-reviews-discovering-sentiment
サンプルコマンド:
Python Visualize.py -seq_length 1000 -cuda -load_model mlstm_ns.pt -temperature 0.4 -neuron 2388 -init「私は理解できなかった」
リリースをクリックして、モデルファイルmlstm_ns.ptまたはnumpyウェイトを取得します。
lm.pyファイルを使用すると、新しいデータでモデルを再操作できます。例えば :
python lm.py -seq_length 50 -batch_size 64 -rnn_size 4096 -embed_size 64 -layers 1 -Learning_rate 0.001 -cuda -load_model mlstm_ns.pt -save_model mlstm -rnn_tpe mlstm -dropout 0 -dropout 0 -dropout 0 -trainデータ/ data/your_validation_file.txt