SenseVoice icon indicating copy to clipboard operation
SenseVoice copied to clipboard

可以返回情感识别结果的置信度吗?

Open lgbest123 opened this issue 1 year ago • 1 comments

Notice: In order to resolve issues more efficiently, please raise issue following the template. (注意:为了更加高效率解决您遇到的问题,请按照模板提问,补充细节)

❓ Questions and Help

使用SenseVoiceSmall模型,如何返回情感识别结果的置信度信息呢?

lgbest123 avatar Oct 06 '24 03:10 lgbest123

ctc_logits = self.ctc.log_softmax(encoder_out) 第2个token为情感概率分布

https://github.com/FunAudioLLM/SenseVoice/blob/969634be261bf30dc3aea3ba317a45d3882f8c52/model.py#L855

gaochangfeng avatar Oct 11 '24 07:10 gaochangfeng