Downloads · 30 days
98
0% of all-time downloads
j-hartmann/purchase-intention-english-roberta-large
purchase-intention-english-roberta-large is a text classification model from j-hartmann. Use it when you need a label for a piece of text. It is set up for transformers.
This RoBERTa-based model can classify expressed purchase intentions in English language text in 2 classes:
Downloads · 30 days
98
0% of all-time downloads
All-time downloads
104K
Public
Repo size
2.9 GB
Likes
6
Public
Click a slice to open those files.
.bin1.4 GB · 100%
From the Hugging Face model README
This RoBERTa-based model can classify expressed purchase intentions in English language text in 2 classes:
The model was fine-tuned on 2,000 manually annotated social media posts. The hold-out accuracy is 95% (vs. a balanced 50% random-chance baseline). For details on the training approach see Web Appendix F in Hartmann et al. (2021).
from transformers import pipeline
classifier = pipeline("text-classification", model="j-hartmann/purchase-intention-english-roberta-large", return_all_scores=True)
classifier("I want this!")
Output:
[[{'label': 'no', 'score': 0.0014553926885128021},
{'label': 'yes', 'score': 0.9985445737838745}]]
Please cite this paper when you use our model. Feel free to reach out to [email protected] with any questions or feedback you may have.
@article{hartmann2021,
title={The Power of Brand Selfies},
author={Hartmann, Jochen and Heitmann, Mark and Schamp, Christina and Netzer, Oded},
journal={Journal of Marketing Research}
year={2021}
}