From e701a672711f4f62a8560182bc92ad48e2c088fc Mon Sep 17 00:00:00 2001 From: blazerye Date: Fri, 26 Jul 2024 11:27:36 +0800 Subject: [PATCH] update --- evaluate/evaluate.md | 2 +- evaluate/main_eval.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/evaluate/evaluate.md b/evaluate/evaluate.md index a933c56..41d6eaf 100644 --- a/evaluate/evaluate.md +++ b/evaluate/evaluate.md @@ -2,7 +2,7 @@ ## Precautions: Our evaluation process adopts the workflow of [ChatDrug](https://github.com/chao1224/ChatDrug) for evaluating small molecules. Therefore, the evaluation code is based on the script provided by this work. Please note that, unlike this work, our success rate calculation method is `the number of successful optimizations` / `the total number of molecules`, while this work is `the number of successful optimizations` / `the number of valid molecules`. -In addition, in our optimization tasks involving "esol", "bbbp", and "herg" properties, the property values are given by the idrug model. Currently, the API call of the idrug model is not open to the public, but the [web query](https://drug.ai.tencent.com/cn) is free for everyone. You can query the aforementioned property values of molecules through this method. If you want to evaluate the above tasks in batches, please implement your own property prediction model at the code entry we reserved. +In addition, in our optimization tasks involving "esol", "bbbp", and "herg" properties, the property values are given by the idrug model. Currently, the API call of the idrug model is not open to the public, but the [web query](https://drug.ai.tencent.com/cn) is available for everyone. You can query the aforementioned property values of molecules through this method. If you want to evaluate the above tasks in batches, please implement your own property prediction model at the code entry we reserved. ## Script Parameter Explanation: `--task`: Task name, currently supported values are: `qed+`, `acceptor+`, `donor+`, `esol+`, `bbbp+`, `herg-`, `esol+acceptor+`, `qed+bbbp+`. diff --git a/evaluate/main_eval.py b/evaluate/main_eval.py index 6745e9e..e4fc888 100644 --- a/evaluate/main_eval.py +++ b/evaluate/main_eval.py @@ -175,14 +175,14 @@ def main_assit(args): parser.add_argument('--num_round', required=False, type=int, default=2, help='number of conversation round') parser.add_argument('--client_add', required=True, type=str, help='DrugAssist client address set by gradio service') parser.add_argument('--database', required=False, type=str, default='mol_DB.csv',help='molecule database') - parser.add_argument('--test_mol', required=False, type=str, default='testset.csv', help='testset ') + parser.add_argument('--test_mol', required=False, type=str, default='testset.csv', help='testset') args = parser.parse_args() args = vars(args) if 'esol' in args['task'] or 'bbbp' in args['task'] or 'herg' in args['task']: choice = input("The property value involved in this task will be given by a specific prediction model, make sure you have implemented it yourself!\ - If you have any questions, please refer to xxx. Enter y to confirm, enter n to exit\n") + If you have any questions, please refer to https://github.com/blazerye/DrugAssist/blob/main/evaluate/evaluate.md. Enter y to confirm, enter n to exit\n") if choice == 'y': main_assit(args) else: