yasufumy / allennlp_imdb

AllenNLP Startup Guide
https://towardsdatascience.com/allennlp-startup-guide-24ffd773cd5b
MIT License
14 stars 12 forks source link
allennlp deep-learning machine-learning natural-language-processing python pytorch

A gentle guild to starting your NLP project with AllenNLP

Requirements

Setup

git clone https://github.com/yasufumy/allennlp_imdb
cd allennlp_imdb
pipenv install
pipenv shell

Usage

There is a post about this repository: Click this link

Running on Colab

Running on CPU:

allennlp train \
    --include-package allennlp_imdb \
    -s /path/to/serialization \
    training_config/base_cpu.jsonnet

Running on GPU:

allennlp train \
    --include-package allennlp_imdb \
    -s /path/to/serialization \
    -o '{"trainer": {"cuda_device": 0}}' \
    training_config/base_cpu.jsonnet

Creating your own configuration file:

allennlp configure --include-package allennlp_imdb