transformer feature importance // 取transformer的特征重要性

耐心是关键

已于 2023-08-15 15:55:48 修改

阅读量371

点赞数

分类专栏：深度学习机器学习文章标签：深度学习机器学习

于 2023-07-31 09:16:17 首次发布

本文链接：https://blog.csdn.net/guotong1988/article/details/132015509

版权

深度学习同时被 2 个专栏收录

216 篇文章 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

机器学习

54 篇文章

订阅专栏

本文探讨了Transformer模型中，每个句子token对预测结果的贡献度，深入理解深度学习中的特征重要性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

即一个句子的每个token的对预测结果的贡献度：

基于 github.com/shap/shap


from transformers import AutoModelForSequenceClassification
from datasets import load_dataset
input_data_dev = load_dataset("data_dir", data_files="data.tsv")

checkpoint_local = "c

了解本专栏

超级会员免费看