cnn lstm bilstm deepcnn clstm in pytorch
version==0.3.1
이것은 내 아키텍처의 버전입니다 --- Pytorch-Text-Classification
텍스트 분류를위한 버트 --- pytorch_bert_text_classification
pyorch : 1.0.1
python : 3.6
torchtext: 0.2.1
cuda : 8.0 (support cuda speed up, can chose, default True)
구성 파일을 수정하고 세부 사항은 구성 디렉토리 (여기)를 참조하십시오.
1、python main.py
2、python main.py --config_file ./Config/config.cfg
3、sh run.sh
Pytorch에 구현 된 일부 신경망이 포함되어 있으며 세부 사항은 모델을 참조하십시오.
SST-1 및 SST-2.
하이퍼 파라미터를 진지하게 조정하지 않았으며 여기에서 열차 로그를 볼 수도 있습니다.
다음 테스트 세트 정확도는 최고의 개발 세트 정확도를 기반으로합니다.
| 데이터/모델 | % SST-1 | % SST-2 |
|---|---|---|
| CNN | 46.1086 | 84.2943 |
| Bi-LSTM | 47.9186 | 86.3262 |
| Bi-Gr | 47.6923 | 86.7655 |