pytorch sentiment neuron

Pytorch-Sentiment-Neuron

Requer Pytorch, Cuda e Python 3.5

Versão pytorch de gerar-reviews-Discovering-Sentiment: https://github.com/openai/generating-reviews-discovering-sentiment

Comando de amostra:

python visualize.py -seq_length 1000 -cuda -load_model mlstm_ns.pt -temperature 0,4 -neuron 2388 -init "eu não consegui descobrir"

Clique em Release para obter o arquivo de modelo MLSTM_NS.PT ou pesos numpy.

O arquivo lm.py permite treinar o modelo em novos dados. Por exemplo :

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_type mlstm -dropout 0 -train data/your_input_file.txt -valid dados/your_validation_file.txt