使用roberta模型无法定义模型.compile或摘要
使用roberta模型无法定义模型.compile或摘要
提问于 2021-06-27 22:43:27
使用罗伯塔模型进行情感分析无法定义模型.compile或摘要
from transformers import RobertaTokenizer, RobertaForSequenceClassification
from transformers import BertConfig
tokenizer = RobertaTokenizer.from_pretrained('roberta-base')
robertamodel = RobertaForSequenceClassification.from_pretrained('roberta-base',num_labels=7)
print('\nBert Model',robertamodel.summary())
loss = tf.keras.losses.SparseCategoricalCrossentropy(from_logits=True)
metric = tf.keras.metrics.SparseCategoricalAccuracy('0accuracy')
optimizer = tf.keras.optimizers.Adam(learning_rate=1e-5,epsilon=1e-08)
robertamodel.compile(loss=loss,optimizer=optimizer,metrics=[metric])
print(robertamodel.summary())我得到了这些错误：'RobertaForSequenceClassification‘对象没有'summary’属性，'RobertaForSequenceClassification‘对象没有'compile’属性
回答 1
Stack Overflow用户
回答已采纳
发布于 2021-06-27 22:59:16
Roberta是基于pytorch的。查看辅助函数TFRobertaModel，将其转换为tensorflow模型。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/68152276
复制相关文章
相似问题
腾讯云开发者
