Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

Appearance settings

harsh-99/Sequential-Data-Loader-and-Model-for-Variable-Length-Data

Open more actions menu

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Sequential-Data-Loader-and-Model-for-Variable-Length-Data

Efficient data loader for text dataset using torch.utils.data.Dataset, collate_fn and torch.utils.data.DataLoader.
Efficient Model for text using torch.nn.utils.rnn.pack_padded_sequence and torch.nn.utils.rnn.pad_packed_sequence.
This Model is used for Sentiment classification on IMDB dataset. For different dataset you have to modify "reader" function in data_loader.py and vocab.py.

Installations Required

Usage

Put the data in the same folder.

  • To create dictionary -:
$ python build_vocab.py
  • To train the model -:
$ python train.py
  • To just see the dataloader functioning -:
Use check_loader.ipynb

About

Data loader and model for variable length data in PyTorch

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published
Morty Proxy This is a proxified and sanitized view of the page, visit original site.