Skip to content

r2llab/wrangl

Repository files navigation

Wrangl

Tests

Fast experiments for NLP and ML. See docs here.

Why?

I built this library to prototype ideas quickly. In essence it combines Hydra and Pytorch Lightning for supervised learning. The following are supported with command line or config tweaks (e.g. no additional boilerplate code):

  • checkpointing
  • early stopping
  • auto git diffs
  • logging to wandb
  • Slurm launcher

Installation

pip install -e .  # add [dev] if you want to run tests and build docs.

# for latest
pip install git+https://github.com/r2llab/wrangl

# pypi release
pip install wrangl

Usage

See the documentation for how to use Wrangl. Examples of projects using Wrangl are found in wrangl.examples. In particular wrangl.examples.learn.xor_clf shows an example of using Wrangl to quickly set up a supervised classification task.

About

Parallel data preprocessing for NLP and ML.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published