Views
No views yet
1from vllm import LLM, SamplingParams
2
3PROMPT_TEMPLATE = "Human:\n{query}\n\n Assistant:\n"
4sampling_params = SamplingParams(temperature=temperature, max_tokens=max_new_token)
5EHR = ['xxxxx1','xxxxx2']
6texts = [PROMPT_TEMPLATE.format(query=k) for k in EHR]
7output = model.generate(texts, sampling_params)EHR = ['慢性乙型肝炎病史10余年,曾有肝功能异常,中医治疗后好转;1年余前查HBsAg转阴,但肝脏病理提示病毒性肝炎伴肝纤维化(G1S3-4)']1res = [
2 { "术语": "慢性乙型肝炎",
3 "语义类型": "疾病、综合征、病理功能",
4 "叙述状态": "存在",
5 "身体部位": "无",
6 "数值": "NA",
7 "单位": "NA",
8 "修饰词": "无" },
9 { "术语": "肝功能异常",
10 "语义类型": "症状、体征、临床所见",
11 "叙述状态": "存在",
12 "身体部位": "无",
13 "数值": "NA",
14 "单位": "NA",
15 "修饰词": "无" },
16 { "术语": "HBsAg",
17 "语义类型": "化学物质、药物",
18 "叙述状态": "不存在",
19 "身体部位": "NA",
20 "数值": "无",
21 "单位": "NA",
22 "修饰词": "NA" },
23{ "术语": "肝脏病理",
24 "语义类型": "诊断操作",
25 "叙述状态": "存在",
26 "身体部位": "无",
27 "数值": "无",
28 "单位": "NA",
29 "修饰词": "NA" },
30 { "术语": "病毒性肝炎",
31 "语义类型": "疾病、综合征、病理功能",
32 "叙述状态": "存在",
33 "身体部位": "无",
34 "数值": "NA",
35 "单位": "NA",
36 "修饰词": "无" },
37 { "术语": "肝纤维化",
38 "语义类型": "疾病、综合征、病理功能",
39 "叙述状态": "存在",
40 "身体部位": "无",
41 "数值": "NA",
42 "单位": "NA",
43 "修饰词": "无" },
44]@misc{ying2025geniegenerativenoteinformation,
title={GENIE: Generative Note Information Extraction model for structuring EHR data},
author={Huaiyuan Ying and Hongyi Yuan and Jinsen Lu and Zitian Qu and Yang Zhao and Zhengyun Zhao and Isaac Kohane and Tianxi Cai and Sheng Yu},
year={2025},
eprint={2501.18435},
archivePrefix={arXiv},
primaryClass={cs.CL},
url={https://arxiv.org/abs/2501.18435},
}