metadata
license: apache-2.0
tags:
- text-classification
- generic
library_name: generic
Hugging Face Transformers with Scikit-learn Classifiers 🤩🌟
This repository contains a small proof-of-concept pipeline that leverages BART embeddings with scikit-learn Logistic Regression that does sentiment analysis. The training leverages the language module of whatlies.
Below you can see the pipeline:
Pipeline(steps=[('embedding',\n HFTransformersLanguage(model_name_or_path='allenai/longformer-base-4096')),\n ('model', LogisticRegression())])Please rerun this cell to show the HTML repr or trust the notebook.
Pipeline(steps=[('embedding',\n HFTransformersLanguage(model_name_or_path='allenai/longformer-base-4096')),\n ('model', LogisticRegression())])
HFTransformersLanguage(model_name_or_path='allenai/longformer-base-4096')
LogisticRegression()