Skip to content

Instantly share code, notes, and snippets.

@prakhar21

prakhar21/flair_sentence_train.py

Created January 19, 2020 17:26
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save prakhar21/cbe435a1c14c4cf9bc528a3cda6c482a to your computer and use it in GitHub Desktop.
Save prakhar21/cbe435a1c14c4cf9bc528a3cda6c482a to your computer and use it in GitHub Desktop.
Download ZIP
Flair Sentence Embeddings and Training a Classifier
Raw
flair_sentence_train.py
document_embeddings = DocumentRNNEmbeddings(word_embeddings, hidden_size=512, reproject_words=True, reproject_words_dimension=256, rnn_type='LSTM', rnn_layers=1, bidirectional=False)
classifier = TextClassifier(document_embeddings, label_dictionary=corpus.make_label_dictionary(), multi_label=False)
trainer = ModelTrainer(classifier, corpus)
trainer.train('./model', max_epochs=20, patience=5, mini_batch_size=32, learning_rate=0.1)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment