Instructions to use Johnade/albert-base-v2-fine-tune-twitter-identity-classification with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Johnade/albert-base-v2-fine-tune-twitter-identity-classification with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="Johnade/albert-base-v2-fine-tune-twitter-identity-classification")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("Johnade/albert-base-v2-fine-tune-twitter-identity-classification") model = AutoModelForSequenceClassification.from_pretrained("Johnade/albert-base-v2-fine-tune-twitter-identity-classification", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Model Description
This project aims to build a model capable of predicting the identity of a twitter user from their twitter.
Before proceeding I would raise the minor caveat that identity here is defined as male, female, or a brand. In some of the code snippets gender is used in place of identity because of how the column was labelled in the data.
Output Key
Label_0 = Brand
Label_1 = Female
Label_2 = Male
Base model
Base model is the mult albert-base-v2
- Downloads last month
- 5