pytorch sentiment neuron

pytorch-neuron-neuron

Nécessite Pytorch, Cuda et Python 3.5

Version Pytorch de la génération de revit-discours-Sentiment: https://github.com/openai/generating-reviews-discoverring-sentiment

Exemple de commande:

python visualize.py -seq_length 1000 -cuda -load_model mlstm_ns.pt -Temperature 0.4-neuron 2388 -Init "Je n'ai pas pu comprendre"

Cliquez sur la version pour obtenir des poids MLSTM_NS.PT ou Numpy Fichier.

Le fichier lm.py vous permet de recycler le modèle sur de nouvelles données. Par exemple :

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. data / your_validation_file.txt