Skip to content

Latest commit

 

History

2 Commits

Folders and files

NameName
Last commit message
Last commit date
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Encoder Is All You Need

Overview

This project implements the Transformer Encoder from the paper "Attention Is All You Need" using PyTorch. The goal is to compute self-attention for multiple heads, combine attention scores, and process sequences efficiently. This implementation is specifically designed for sentiment analysis on the IMDB dataset.

Features

  • ✅ Multi-Head Self-Attention
  • ✅ Positional Encoding
  • ✅ Layer Normalization
  • ✅ Feedforward Networks
  • ✅ Scaled Dot-Product Attention
  • ✅ Sentiment Classification
  • ✅ Configurable Hyperparameters

📊 Dataset

This project uses the IMDB movie reviews dataset for sentiment analysis. The dataset consists of 50,000 reviews, labeled as either positive or negative.

You can download the dataset using:

from torchtext.datasets import IMDB
train_iter, test_iter = IMDB()

📚 References


💡 For any issues, feel free to open an issue or contribute! 🚀

About

No description, website, or topics provided.

Resources

Stars

0 stars

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages