From 6ffa75ea34b47017aa1a9ba909ed304cb56ba254 Mon Sep 17 00:00:00 2001
From: SeonghwanSeo <shwan0106@gmail.com>
Date: Mon, 19 Jun 2023 00:31:18 +0900
Subject: [PATCH] add mw-logp

---
 download-weights.sh                 | 16 +++++++++++-----
 test/generation_config/mw_logp.yaml |  8 ++++++++
 2 files changed, 19 insertions(+), 5 deletions(-)
 create mode 100644 test/generation_config/mw_logp.yaml

diff --git a/download-weights.sh b/download-weights.sh
index 2d36355..a766d90 100644
--- a/download-weights.sh
+++ b/download-weights.sh
@@ -1,5 +1,11 @@
-conda install --yes -c conda-forge gdown
-cd ./test
-gdown '16BK_tjOUmuMuWjSvV3TeLVc86PeAKig-'
-tar -xvf 'pretrained_model.tar'
-cd ../
+#conda install --yes -c conda-forge gdown
+mkdir -p ./test/pretrained_model
+cd ./test/pretrained_model
+gdown 'https://drive.google.com/uc?id=1t-gS7bCOxO5gqv8GXkluTzx1mO20_L0G'    # logp
+gdown 'https://drive.google.com/uc?id=10UXskmoQQW6KZAeNSVqF4LliMZB9j5Iq'    # mw
+gdown 'https://drive.google.com/uc?id=1897Gpa27jFOAYcGupdVGsPpMMbqDgcU9'    # qed
+gdown 'https://drive.google.com/uc?id=1kRqaLIl152vw1yGzS2Dji3jPas9ksW0V'    # tpsa
+gdown 'https://drive.google.com/uc?id=1udYJuz_p3pMxGBWsJyqHMcrT1l7saUkB'    # 3cl_affinity
+gdown 'https://drive.google.com/uc?id=1lY5gU6YBDSZvqvEosEPlQ1w3HTe55sCJ'    # logp_tpsa
+gdown 'https://drive.google.com/uc?id=1uFJbHMCRlXOTDIzWK-Dn6z8R9SeafDDx'    # mw_logp
+cd ../../
diff --git a/test/generation_config/mw_logp.yaml b/test/generation_config/mw_logp.yaml
new file mode 100644
index 0000000..843b55b
--- /dev/null
+++ b/test/generation_config/mw_logp.yaml
@@ -0,0 +1,8 @@
+model_path: ./test/pretrained_model/mw_logp.tar
+library_path: ./data/ZINC/library.csv
+library_builtin_model_path: ./test/builtin_model/mw_logp.tar 
+
+# Required
+window_size: 2000
+alpha: 0.75
+max_iteration: 10