mirror of
https://github.com/modelscope/modelscope.git
synced 2026-09-01 19:49:03 +02:00
[to #42322933] add UT for adaseq span-based ner pipeline
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11398124
This commit is contained in:
0
tests/pipelines/adaseq_pipelines/__init__.py
Normal file
0
tests/pipelines/adaseq_pipelines/__init__.py
Normal file
@@ -0,0 +1,25 @@
|
||||
# Copyright (c) Alibaba, Inc. and its affiliates.
|
||||
import os
|
||||
import unittest
|
||||
|
||||
from modelscope.pipelines import pipeline
|
||||
from modelscope.utils.constant import Tasks
|
||||
from modelscope.utils.demo_utils import DemoCompatibilityCheck
|
||||
from modelscope.utils.test_utils import test_level
|
||||
|
||||
|
||||
class NamedEntityRecognitionTest(unittest.TestCase, DemoCompatibilityCheck):
|
||||
|
||||
def setUp(self):
|
||||
os.system('pip install adaseq>=0.5.0')
|
||||
|
||||
@unittest.skipUnless(test_level() >= 0, 'skip test in current test level')
|
||||
def test_run_span_based_ner_pipeline(self):
|
||||
pipeline_ins = pipeline(
|
||||
Tasks.named_entity_recognition,
|
||||
'damo/nlp_nested-ner_named-entity-recognition_chinese-base-med')
|
||||
print(
|
||||
pipeline_ins(
|
||||
'1、可测量目标: 1周内胸闷缓解。2、下一步诊疗措施:1.心内科护理常规,一级护理,低盐低脂饮食,留陪客。'
|
||||
'2.予“阿司匹林肠溶片”抗血小板聚集,“呋塞米、螺内酯”利尿减轻心前负荷,“瑞舒伐他汀”调脂稳定斑块,“厄贝沙坦片片”降血压抗心机重构'
|
||||
))
|
||||
Reference in New Issue
Block a user