search

GDGVIT / quicktext

Toolkit for text classification
https://picturate.github.io/quicktext/
MIT License
3 stars 4 forks source link

Add base model for text classifiers #21

Closed ramaneswaran closed 4 years ago

ramaneswaran commented 4 years ago

🚀 Feature

Add a base model which inherits pl.Lightningmodule. This class contains optimizers, loss function (cross-entropy loss) and the training and validation steps

Users can inherit this class and override the forward method to create new models

ramaneswaran commented 4 years ago

Use this class for BiLSTM model too