Adfluence-AI Purchase Intent Classifier
This model is a fine-tuned version of Davlan/afro-xlmr-base, a powerful multilingual model with a strong understanding of African languages. It has been specifically trained to classify purchase intent in social media comments written in Amharic (Ge'ez script), Romanized Amharic, and mixed Amharic-English (code-switching).
This model was developed for the Adfluence AI project, which aims to evaluate the effectiveness of influencer marketing campaigns in Ethiopia.
It achieves a weighted F1-score of 0.81 on the evaluation set.
Model Description
The model takes a social media comment as input and outputs a prediction across five categories of purchase intent:
highly_likely
likely
neutral
unlikely
highly_unlikely
This allows for a nuanced understanding of audience reaction beyond simple positive/negative sentiment, directly measuring the potential for user conversion.
How to Use
You can use this model directly with the pipeline
function from the transformers
library.
from transformers import pipeline
# Load the model from the Hub
model_id = "YosefA/adfluence-intent-model"
classifier = pipeline("text-classification", model=model_id)
# --- Example Usage ---
# Example 1: Amharic (Ge'ez Script) - Clear intent
comment_1 = "แแ แ แฃแ แ แชแ แแ! แจแต แแ แแแแต แจแแฝแแ?"
# Translation: "Wow, this is great! Where can I find it?"
# Example 2: Mixed Amharic-English - Neutral/Questioning
comment_2 = "Hmm, interesting. Price-u endet new?"
# Translation: "Hmm, interesting. How is the price?"
# Example 3: Romanized Amharic - Negative
comment_3 = "Ene enja minim altemechegnim, quality yelelew neger new."
# Translation: "I don't know, I didn't like it at all, it's a thing with no quality."
results = classifier([comment_1, comment_2, comment_3])
for comment, result in zip([comment_1, comment_2, comment_3], results):
print(f"Comment: {comment}")
print(f"Prediction: {result['label']}, Score: {result['score']:.4f}\n")
# Expected Output:
# Comment: แแ แ แฃแ แ แชแ แแ! แจแต แแ แแแแต แจแแฝแแ?
# Prediction: highly_likely, Score: 0.9851
#
# Comment: Hmm, interesting. Price-u endet new?
# Prediction: neutral, Score: 0.9214
#
# Comment: Ene enja minim altemechegnim, quality yelelew neger new.
# Prediction: highly_unlikely, Score: 0.9902
Intended Uses & Limitations
Intended Use
This model is intended to be used as a backend component for the Adfluence AI platform. Its primary purpose is to analyze user comments on social media advertisements (e.g., on Instagram, Facebook, TikTok) to gauge audience purchase intent and provide campaign performance metrics.
Limitations
- Simulated Data: The model is trained on a high-quality simulated dataset, not on live social media data. While designed to reflect real-world usage, performance may vary on wild, un-sanitized data.
- Domain Specificity: The source data was derived from product reviews (specifically for electronics). The model's performance may be strongest in the e-commerce/product domain and might require further fine-tuning for vastly different domains like services, events, or fashion.
- Language Scope: The model only understands Amharic and English. It has not been trained on other Ethiopian languages like Tigrinya, Oromo, etc.
Training and Evaluation Data
This model was fine-tuned on the custom YosefA/Adflufence-ad-comments
dataset.
The dataset was created through the following process:
- Source: Started with ~3180 English product reviews from an Amazon dataset.(https://huggingface.co/datasets/hugginglearners/amazon-reviews-sentiment-analysis)
- Transformation: Each review was programmatically rephrased and translated into a simulated social media comment using Google's Gemini Flash.
- Stylization: Comments were generated in three styles to mimic real-world Ethiopian user behavior:
- Amharic (Geโez script)
- Romanized Amharic
- Mixed Amharic-English (Code-Switching)
- Enrichment: Comments were styled with emojis, slang, and informal sentence structures.
- Labeling: Each comment was assigned a purchase intent label mapped from the original star rating of the source review.
Training Procedure
Training Hyperparameters
The following hyperparameters were used during training:
learning_rate
: 2e-05train_batch_size
: 16eval_batch_size
: 16seed
: 42optimizer
: AdamW with betas=(0.9,0.999) and epsilon=1e-08lr_scheduler_type
: linearnum_epochs
: 3
Training Results
The model achieved its best performance at the end of Epoch 2.
Training Loss | Epoch | Step | Validation Loss | F1 (Weighted) |
---|---|---|---|---|
No log | 1.0 | 160 | 0.5001 | 0.7852 |
No log | 2.0 | 320 | 0.4316 | 0.8101 |
No log | 3.0 | 480 | 0.4281 | 0.8063 |
Framework Versions
- Transformers 4.41.2
- Pytorch 2.3.0+cu121
- Datasets 2.19.0
- Tokenizers 0.19.1
- Downloads last month
- 20
Model tree for YosefA/adfluence-intent-model
Base model
Davlan/afro-xlmr-baseDataset used to train YosefA/adfluence-intent-model
Space using YosefA/adfluence-intent-model 1
Evaluation results
- F1 (Weighted) on YosefA/Adflufence-ad-commentstest set self-reported0.810