Skip to content

A tutorial on RNNs and CNNs in PyTorch using a simple name classification example

License

Notifications You must be signed in to change notification settings

bobflagg/classifying-names

Repository files navigation

Classifying Names with RNNs and CNNs in PyTorch

This repo contains a tutorial on RNNs and CNNs in PyTorch based on the name classification with RNNs example by Sean Robertson. In that tutorial, the RNN is implemented from scratch. Here I'll use PyTorch's build-in RNN and LSTM implementations and compare results with a CNN classifier implemented in PyTorch and a simple Naive Bayes classifier implemented in scikit-learn. Please note that this tutorial is not done and may contain many errors! I'm making the incomplete code available as I finish it in-case it may be useful. Caveat emptor!

  1. Preparing the Data
  2. Classifying with an RNN
  3. Classifying with an LSTM
  4. Classifying with a CNN

About

A tutorial on RNNs and CNNs in PyTorch using a simple name classification example

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published