模型:

j-hartmann/purchase-intention-english-roberta-large

英文

这个基于RoBERTa的模型可以对英文文本中的购买意图进行分类,分为2个类别:

  • 购买意图 ?
  • 无购买意图 ?

该模型在2,000个人工注释的社交媒体帖子上进行了微调。保留准确率为95%(相对于平衡的50%随机基线)。有关训练方法的详细信息,请参阅Hartmann等人的网络附录 F(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}]]

参考

在使用我们的模型时,请引用 this paper 。如果您有任何问题或反馈,请随时通过jochen.hartmann@tum.de与我们联系。

@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}
}