ai_tweet_classifier

<b> MULTI LABEL Tweet Classification : Classify tweets into multiple Artificial Intelligence related Categories </b>

This respository contains a bert model finetuned to classify a tweet/linkedin post. The model is trained as a multi label predictor, which will return multiple categories for each input. Model is trained on following 8 categories :

'AI News', 'AI Tools', 'AI Research', 'AI Models', 'AI Usecases','AI Open Source', 'Podcasts/Talks/Events', 'AI Opinions','Non AI'

<b>Usage: </b>

<code >

model = AutoModelForSequenceClassification.from_pretrained( model_path, )

tokenizer = AutoTokenizer.from_pretrained( model_path)

from transformers import pipeline

classify_tweet = pipeline("text-classification", model=model,tokenizer = tokenizer,return_all_scores=True)

output = classify_tweet (your text)

</code>

<b> Labels used: </b>

<ul> <li> LABEL_0 = 'AI News' </li> <li> LABEL_1 = 'AI Tools'</li> <li> LABEL_2 = 'AI Research'</li> <li> LABEL_3 = 'AI Models'</li> <li> LABEL_4 = 'AI Usecases'</li> <li> LABEL_5 = 'AI Open Source'</li> <li> LABEL_6 = 'Podcasts/Talks/Events'</li> <li> LABEL_7 = 'Opinions'</li> <li> LABEL_8 = 'Non AI' </li> </ul>