ersilia-os/ersilia
The Ersilia Model Hub, an open-source repository and CLI of AI/ML models for infectious and neglected disease research.
什么是 Ersilia?
Ersilia 是一个命令行工具和 Python 库,允许你下载、运行和管理用于药物发现相关任务(例如预测抗生素活性、ADMET 性质、分子表征、生成化学)的预训练 AI/ML 模型。所有模型都位于 Ersilia Model Hub 中,这是一个专注于传染病和被忽视疾病的精选模型目录。
工作原理
- 安装 CLI(通过 Conda、
pip或直接克隆仓库)。CLI 是所有操作的核心入口。 - 获取模型 – 每个模型都有一个短标识符(例如
eos4e40)。CLI 从 hub 中拉取模型的容器(基于 Docker 或 Conda)并将其存储在本地。 - 提供模型 – 工具启动一个轻量服务(Docker 容器或 Conda 环境),暴露模型的推理 API。
- 运行推理 – 向服务提供 CSV(或 SMILES 字符串列表),并在 CSV/HDF5 文件或 pandas DataFrame 中接收预测结果。
- 管理生命周期 – 你可以列出模型、查看信息、测试、关闭服务或删除本地副本。
同样的工作流也可以通过 Python 中的 ersilia.api 模块获得,因此脚本可以以编程方式获取、提供和运行模型。
谁会使用它?
- 研究人员(学术界或工业界),需要快速访问经验证的抗菌或其他疾病相关化合物的预测模型。
- 生物信息学家,希望将模型嵌入管道而无需处理复杂的环境设置。
- 教育工作者/学生,寻找 AI 驱动的药物发现模型的现成示例。
主要优势
| 方面 | 重要性 |
|---|---|
| 低代码/CLI | 无需编写样板推理代码;一条命令即可获取、提供和运行模型。 |
| 容器化模型 | 每个模型在独立的隔离环境(Docker 或 Conda)中运行,避免依赖冲突。 |
| 领域专注的目录 | 所有模型针对传染病或被忽视疾病,并附带清晰的引用和来源。 |
| Python API | 无缝集成到笔记本或更大的管道中。 |
| 开源且可复现 | GPL-3 许可证,PyPI/conda-forge 上的版本化发布,以及用于引用的 DOI。 |
限制
- hub 仅包含 Ersilia 社区贡献的模型;不遵循其模板工作流,你无法上传任意模型。
- 运行模型需要 Docker(或 Conda)以及足够的计算资源来支撑底层 ML 模型;非常大的模型在普通硬件上可能运行缓慢。
- CLI 专注于批量 CSV 输入;交互式或实时推理需要自定义脚本。
快速开始(CLI)
# 1. 安装 CLI
conda create -n ersilia python=3.10 && conda activate ersilia
pip install ersilia # 或克隆后执行 pip install -e .
# 2. 拉取模型
ersilia fetch eos4e40
# 3. 启动服务
ersilia serve eos4e40
# 4. 在 CSV 上运行预测
ersilia run -i my_input.csv -o my_output.csv
# 5. 清理
ersilia close
ersilia delete eos4e40
快速开始(Python)
from ersilia.api import Model
m = Model('eos4e40')
m.fetch()
m.serve()
results = m.run(['CCO', 'c1ccccc1']) # pandas DataFrame
m.close()
结论: Ersilia 提供了一个即用型、低代码的接口,用于访问针对传染病药物发现的精选 AI 模型集合。它抽象了环境管理,让科学家专注于应用模型而不是从头构建模型。
相关
- 项目
- 项目
- 项目
- 项目
- 项目