开发者社区 > ModelScope模型即服务 > 计算机视觉 > 正文

ModelScope这个模型可以设置输出的阈值?

ModelScope这个模型可以设置输出的阈值?image.png

展开
收起
真的很搞笑 2024-01-02 20:15:34 66 0
2 条回答
写回答
取消 提交回答
  • 面对过去,不要迷离;面对未来,不必彷徨;活在今天,你只要把自己完全展示给别人看。

    是的,ModelScope确实支持设置输出阈值。具体来说,你可以在每次推理请求中添加top-k或者top-p参数来控制模型生成的输出结果。这些参数可以通过设置相应的请求头或者请求参数来实现。例如,"top-k"参数可以指定从生成概率值最高的k个候选项中进行选择。通过这种方式,你可以根据需要调整输出结果的精度和数量,进一步优化你的应用体验。

    2024-01-03 13:39:58
    赞同 展开评论 打赏
  • 在ModelScope中,你可以通过设置模型的参数来控制输出的阈值。例如,对于分类模型,你可以设置阈值来决定哪些类别是阳性的,哪些类别是阴性的。

    在你的例子中,你正在使用一个基于DeBERTa的命名实体识别模型。这个模型的输出是一个包含命名实体及其类别的列表。你可以通过设置阈值来决定哪些实体被认为是重要的,哪些实体被认为是无关的。

    以下是一个使用ModelScope的示例,展示了如何设置模型的参数:

    from modelscope.pipelines import pipeline
    from modelscope.utils.constant import Tasks
    
    semantic_cls = pipeline('rex-uninlu', model='damo/nlp_deberta_rex-uninlu_chinese-base', model_revision='v1.2.1')
    
    # 设置阈值为0.5,即当模型预测的置信度大于0.5时,认为实体是阳性的
    semantic_cls.set_threshold(0.5)
    
    # 使用模型进行预测
    predictions = semantic_cls('这是一段文本,其中包含一个实体:阿里巴巴。')
    
    # 输出预测结果
    print(predictions)
    

    在这个示例中,set_threshold()方法设置了模型的阈值为0.5。然后,pipeline()方法使用模型进行预测,并输出预测结果。

    希望这些信息对你有所帮助。

    2024-01-03 09:47:24
    赞同 展开评论 打赏

包含图像分类、图像生成、人体人脸识别、动作识别、目标分割、视频生成、卡通画、视觉评价、三维视觉等多个领域

热门讨论

热门文章

相关电子书

更多
视觉AI能力的开放现状及ModelScope实战 立即下载
ModelScope助力语音AI模型创新与应用 立即下载
低代码开发师(初级)实战教程 立即下载