需要安装transformers
、shap
、matplotlib
、ipython
等库。
Pycharm会显示
<IPython.core.display.HTML object>
,只能在jupyter notebook中正常显示。
import shap
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import transformers
model_dir = "roberta/trained_model"
tokenizer = AutoTokenizer.from_pretrained(model_dir)
model = AutoModelForSequenceClassification