pytorch_RVAE

Pytorch 재발 변형 autoencoder

모델:

이것은 Kim의 캐릭터 인식 신경 언어 모델이 토큰을 포함하는 연속 공간에서 Samuel Bowman의 문장을 구현하는 것입니다.

샘플링 예 :

the new machine could be used to increase the number of ventures block in the company 's <unk> shopping system to finance diversified organizations

미국 공무원들은 또한 제안이 촉구 및 프로그램으로 사용될 수 있는지 여부를 기꺼이 고려할 것이라고 말했다.

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

회사는 컬러 세트가 프로그램에 포함될 수 있다고 주장했다.

용법

모델 훈련 전에 단어 임베딩을 훈련시켜야합니다.

 $ python train_word_embeddings.py

이 스크립트는 Mikolov et al. 단어와 문구의 분산 된 표현

매개 변수 :

--use-cuda

--num-iterations

--batch-size

--num-sample –– 소음 토큰에서 샘플링 된 수

모델 사용 훈련 :

 $ python train.py

매개 변수 :

--use-cuda

--num-iterations

--batch-size

--learning-rate

--dropout –– 디코더 입력에서 제로화 될 확률

--use-trained --모델 이전 모델을 사용하십시오

교육 사용 후 데이터를 샘플링하려면 :

 $ python sample.py

매개 변수 :

--use-cuda

--num-sample