标签： deep-learning nlp text-classification bert-language-model huggingface-transformers
这里需要一些帮助： 我已经训练了我的XLMR模型进行分类，现在我正在尝试完善测量集的输出。
我已经使用加载了模型
`encoder = TFModel.from_pretrained(pt_teacher_checkpoint, from_pt=True,config=config)' 'embedding = encoder(input_ids)`
如何从嵌入中获取类（在我的情况下为sst-2）/最后一层输出（在我的情况下为（batch_size，2））？ 有人可以帮我吗？