cnn lstm bilstm deepcnn clstm in pytorch
version==0.3.1
Ceci est une version de ma propre architecture --- pytorch-text-classification
Bert pour la classification du texte --- 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)
Modifiez le fichier de configuration, consultez le répertoire de configuration (ici) pour plus de détails.
1、python main.py
2、python main.py --config_file ./Config/config.cfg
3、sh run.sh
Contient quelques réseaux de neurones implémentaires à Pytorch, voir les modèles pour les détails.
SST-1 et SST-2.
Je n'ai pas ajusté sérieusement les hyper-paramètres, vous pouvez également voir le train se connecter ici.
La précision des tests de test suivants est basée sur la meilleure précision de jeu de développement.
| Données / modèle | % SST-1 | % SST-2 |
|---|---|---|
| CNN | 46.1086 | 84.2943 |
| Bi-lstm | 47.9186 | 86.3262 |
| Bi-gru | 47.6923 | 86.7655 |