首页
/ icml18-jtnn 项目使用教程

icml18-jtnn 项目使用教程

2024-09-25 22:36:26作者:胡易黎Nicole

1. 项目目录结构及介绍

icml18-jtnn/
├── bo/
│   ├── README.md
│   └── ...
├── data/
│   └── ...
├── fast_jtnn/
│   ├── README.md
│   └── ...
├── fast_molvae/
│   ├── README.md
│   └── ...
├── jtnn/
│   ├── README.md
│   └── ...
├── molopt/
│   ├── README.md
│   └── ...
├── molvae/
│   ├── README.md
│   └── ...
├── .gitignore
├── LICENSE
├── README.md
├── paradigm.png
└── requirements.txt

目录结构介绍

  • bo/: 包含用于贝叶斯优化实验的脚本。请阅读 bo/README.md 获取详细信息。
  • data/: 数据文件夹,包含项目所需的数据。
  • fast_jtnn/: 包含模型实现的代码。
  • fast_molvae/: 包含用于VAE训练的代码。请阅读 fast_molvae/README.md 获取详细信息。
  • jtnn/: 包含模型公式的代码。
  • molopt/: 包含用于联合训练VAE和属性预测器的脚本。请阅读 molopt/README.md 获取详细信息。
  • molvae/: 包含仅用于训练VAE模型的脚本。请阅读 molvae/README.md 获取详细信息。
  • .gitignore: Git忽略文件。
  • LICENSE: 项目许可证文件。
  • README.md: 项目主文档。
  • paradigm.png: 项目图示文件。
  • requirements.txt: 项目依赖文件。

2. 项目启动文件介绍

项目的启动文件主要集中在 fast_molvae/ 目录下。具体启动文件为 fast_molvae/train.py。该文件用于启动VAE模型的训练过程。

启动步骤

  1. 进入 fast_molvae/ 目录:

    cd fast_molvae/
    
  2. 运行训练脚本:

    python train.py
    

3. 项目的配置文件介绍

项目的配置文件主要为 requirements.txt,该文件列出了项目运行所需的依赖包。

requirements.txt 内容

rdkit
numpy
torch

安装依赖

在项目根目录下运行以下命令安装依赖:

pip install -r requirements.txt

以上步骤完成后,您就可以顺利启动并运行 icml18-jtnn 项目。

登录后查看全文
热门项目推荐
相关项目推荐