robinvanschaik / interpret-flair

A small repository to test Captum Explainable AI with a trained Flair transformers-based text classifier.
MIT License
24 stars 2 forks source link

Add support for other type of Flair models #3

Open robinvanschaik opened 3 years ago

robinvanschaik commented 3 years ago

Flair offers multiple types of text classification models.

From transformers-based models via huggingface (currently implemented) to word embeddings.

Restructuring to allow for other types models would be nice.