Pytorch Embedding Model Part 2

Stephen Blum · Beginner ·📐 ML Fundamentals ·4d ago

Key Takeaways

Implements a word embedding model in PyTorch using tokenization and dictionary-based indexing

Original Description

I am pumped today because we finally have a solid plan for our word embedding model, and I want to build it in PyTorch. The goal is simple. Take words, tokenize them into dictionary indexes, pad the context, and turn inputs into vectors we can train on. During training, we will predict the next word from the last three words, like found the bug leads to win, then the bug win leads to you. We hit a few snags, like a regex that did not clean punctuation and some tensor type errors, but we fixed them. We also replaced torch.nn.Embedding with our own embedding tensor and learned we can just index rows instead of doing full one hot matmul. Next up is a small data iterator for batching, then cross entropy loss, then training, and finally testing word similarity. We committed the code and we will start training tomorrow.
Watch on YouTube ↗ (saves to browser)
Sign in to unlock AI tutor explanation · ⚡30

Related Reads

📰
Introduction Data Science and Machine Learning
Learn the basics of data science and machine learning to stay ahead in the emerging field
Medium · Data Science
📰
AgriScore: An Explainable AI Credit Scoring System for Smallholder Farmers
Learn how AgriScore uses explainable AI for credit scoring to help smallholder farmers access loans, and why this matters for financial inclusion
Medium · Machine Learning
📰
AgriScore: An Explainable AI Credit Scoring System for Smallholder Farmers
Learn how AgriScore uses explainable AI for credit scoring of smallholder farmers, and how to build a similar system using machine learning and Streamlit
Medium · Data Science
📰
The Sophistication Trap: Why the Smarter AI Technique Keeps Losing
Smarter AI techniques don't always guarantee better results, and understanding the sophistication trap can help you optimize your approach
Medium · Machine Learning
Up next
How to start learning AI | Complete AI Learning Path | Roadmap For Beginners (With No Background)
Career Talk
Watch →