Skip to content

Sentence representation with CBOW, Relation Network, CNN, Self Attention using PyTorch

Notifications You must be signed in to change notification settings

goddoe/text-classification-and-sentence-representation

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

21 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Text Classification and Sentence Representation

PyTorch implementation of several sentence representation methods: CBOW, Relation Network(RN), CNN, Self Attention.

Contents

  • CBOW
  • Relation Network (RN)
  • CNN
  • Self Attention

Experiments

Experiments without fine-tuning.

Common Settings.

  • Dataset: IMDB
  • learning rate: 0.001
Model Epochs Train Accuracy Test Accuracy
CBOW 5 0.95 0.89
RN 5 1.0 0.78
CNN 5 1.0 0.82

About

Sentence representation with CBOW, Relation Network, CNN, Self Attention using PyTorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages