Thanks! I want to do something similar to what is discussed here: How to build a Text-to-Feature Extractor based on Fine-Tuned BERT Model · Issue #1323 · huggingface/transformers · GitHub 41
But instead of using the text features as input in a classification problem I want to use them as input for regression problem.