Allow CNN/LSTM to get chosen for ngrams from the command line

GateNLP / gate-lf-pytorch-json

PyTorch wrapper for the LearningFramework GATE plugin

Apache License 2.0

1 stars 2 forks source link

Allow CNN/LSTM to get chosen for ngrams from the command line #33

Closed johann-petrak closed 5 years ago

johann-petrak commented 5 years ago

Add a command line argument to choose cnn vs. lstm for ngrams in the classification module. Maybe also allow then to set the most important parameters for the CNN or LSTM. Alternately, just provide a pre-implemented CNN and LSTM module in the modules dir which is easy to update.

johann-petrak commented 5 years ago

Implement command line argument/config setting ngram_layer for this. Fixed as of 2f8428b0