pytorch_RVAE

Pytorch AutoEncoder variacional recorrente

Modelo:

Esta é a implementação das frases geradoras de Samuel Bowman de um espaço contínuo com os modelos de linguagem neural de Kim, conscientes de caráter, incorporando para tokens

Exemplos de amostragem:

A nova máquina pode ser usada para aumentar o número de empreendimentos no sistema comercial da empresa para financiar organizações diversificadas

Funcionários do governo dos EUA também disseram que estariam dispostos a considerar se a proposta poderia ser usada como insistência e programas

men believe they had to go on the <unk> because their <unk> were <unk> expensive important

As empresas insistiram que o conjunto de cores poderia ser incluído no programa

Uso

Antes do treinamento do modelo, é necessário treinar incorporações de palavras:

 $ python train_word_embeddings.py

Esta palavra do trem de script incorporação definida em Mikolov et al. Representações distribuídas de palavras e frases

Parâmetros:

--use-cuda

--num-iterations

--batch-size

--num-sample .

Para treinar o uso do modelo:

 $ python train.py

Parâmetros:

--use-cuda

--num-iterations

--batch-size

--learning-rate

--dropout -Probabilidade de unidades a se zero em entrada de decodificador

--use-trained -use treinado antes do modelo

Para provar dados após o uso do treinamento:

 $ python sample.py

Parâmetros:

--use-cuda

--num-sample