🚀 JSL-MedMNX-7B
JSL-MedMNX-7B 是一款由 John Snow Labs 开发的拥有 70 亿参数的模型。该模型在医学数据集上进行训练,旨在为生物医学基准测试提供最先进的性能,可参考 Open Medical LLM Leaderboard。
此模型遵循 CC-BY-NC-ND 许可协议,同时必须遵守 Acceptable Use Policy。如果您需要将此模型用于商业用途,请通过 info@johnsnowlabs.com 与我们联系。
🚀 快速开始
安装依赖
!pip install -qU transformers accelerate
代码示例
from transformers import AutoTokenizer
import transformers
import torch
model = "johnsnowlabs/JSL-MedMNX-7B"
messages = [{"role": "user", "content": "What is a large language model?"}]
tokenizer = AutoTokenizer.from_pretrained(model)
prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
pipeline = transformers.pipeline(
"text-generation",
model=model,
torch_dtype=torch.float16,
device_map="auto",
)
outputs = pipeline(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
print(outputs[0]["generated_text"])
🏆 评估结果
任务评估
任务 |
版本 |
过滤器 |
n-shot |
指标 |
值 |
|
标准误差 |
stem |
N/A |
none |
0 |
acc_norm |
0.5191 |
± |
0.0068 |
|
|
none |
0 |
acc |
0.5658 |
± |
0.0058 |
- medmcqa |
Yaml |
none |
0 |
acc |
0.5135 |
± |
0.0077 |
|
|
none |
0 |
acc_norm |
0.5135 |
± |
0.0077 |
- medqa_4options |
Yaml |
none |
0 |
acc |
0.5373 |
± |
0.0140 |
|
|
none |
0 |
acc_norm |
0.5373 |
± |
0.0140 |
- anatomy (mmlu) |
0 |
none |
0 |
acc |
0.6370 |
± |
0.0415 |
- clinical_knowledge (mmlu) |
0 |
none |
0 |
acc |
0.7245 |
± |
0.0275 |
- college_biology (mmlu) |
0 |
none |
0 |
acc |
0.7500 |
± |
0.0362 |
- college_medicine (mmlu) |
0 |
none |
0 |
acc |
0.6590 |
± |
0.0361 |
- medical_genetics (mmlu) |
0 |
none |
0 |
acc |
0.7200 |
± |
0.0451 |
- professional_medicine (mmlu) |
0 |
none |
0 |
acc |
0.7206 |
± |
0.0273 |
- pubmedqa |
1 |
none |
0 |
acc |
0.7720 |
± |
0.0188 |
分组评估
分组 |
版本 |
过滤器 |
n-shot |
指标 |
值 |
|
标准误差 |
stem |
N/A |
none |
0 |
acc_norm |
0.5191 |
± |
0.0068 |
|
|
none |
0 |
acc |
0.5658 |
± |
0.0058 |
📄 许可证
本项目采用 CC-BY-NC-ND 4.0 许可协议。