chachong/co_rom.py

import os
os.environ['KMP_DUPLICATE_LIB_OK']='True'


# 方法1
from modelscope.pipelines import pipeline
from modelscope.utils.constant import Tasks
# damo/nlp_corom_sentence-embedding_chinese-base-ecom
# damo/nlp_corom_sentence-embedding_chinese-base
# damo/nlp_corom_sentence-embedding_chinese-tiny
model_id = "damo/nlp_corom_sentence-embedding_chinese-base-ecom"
# model_id = "damo/nlp_gpt3_text-generation_chinese-base"
pipeline_se = pipeline(Tasks.sentence_embedding, model=model_id, max_length=1024)
#
# 当输入包含“soure_sentence”与“sentences_to_compare”时，会输出source_sentence中首个句子与sentences_to_compare中每个句子的向量表示，以及source_sentence中首个句子与sentences_to_compare中每个句子的相似度。
sentences_to_compare = [
            '''根据不同的季度、网红热点信息主动给用户推送热门景区和景点的游玩信息''',
            '''开发一个在线商品销售商城，提供旅游路线上的相关特产和旅游商品的查阅及下单功能''',
            '''推送相关旅游产品折扣信息以及景区景点举办的篝火晚会、烟花盛宴等活动内容，可通过线下报名参与。'''
        ]
inputs = {
        "source_sentence": [
            # '''模块功能：提供商家信息查询和种植区域查询等功能。功能描述：商家信息包括名称、地址、法人、联系方式、经营范围、投资规模等，种植区域主要是指丽水香茶种植区域的面积、位置、高程等信息，政府工作人员可根据多个查询条件完成上述信息的查询。'''
            '''展示乡村游中相关乡村举办的庆典活动，包含庆典举办时间、内容等'''
        ],
        "sentences_to_compare": sentences_to_compare
    }

result = pipeline_se(input=inputs)
print(result["scores"])

arr = result["scores"]
max_value = max(arr)
max_index = arr.index(max_value)
print("最大值：", max_value)
print("最相识内容：", sentences_to_compare[max_index])
# #
#