
SchNetPack:适用于原子系统的深度神经网络工具包
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
简介:SchNetPack是一款针对原子系统设计的深度学习框架,旨在简化分子与材料科学中基于神经网络模型的研究和应用开发。
SchNetPack旨在为原子系统提供易于使用的深度神经网络模型,这些模型能够直接用于训练与应用,并且支持自定义架构的扩展。当前提供的模型包括:
- SchNet:一种端到端连续过滤器CNN,适用于分子和材料。
- wACSF:原子为中心的加权对称函数。
我们将继续努力改进文档、增加更多数据集的支持及提供更多的功能选项。
安装要求:
- Python3
- 日月光麻木PyTorch(> = 0.4.1)
- h5py
可选依赖项:
- tensorboardX
注意:我们建议使用GPU进行神经网络的训练。
可以通过pip命令直接安装SchNetPack:
```
pip install schnetpack
```
或者从源代码进行安装,步骤如下:
克隆GitHub上的存储库并进入目录
```shell
git clone https://github.com/atomistic-machine-learning/schnetpack.git
cd schnetpack
```
全部评论 (0)
还没有任何评论哟~


