adding the model with 4 encoders

PNNL-CompBio · Feb 20, 2024 · 38c7076 · 38c7076
1 parent 121230b
commit 38c7076
Show file tree

Hide file tree

Showing 24 changed files with 2,808 additions and 0 deletions.
diff --git a/model4/README.md b/model4/README.md
@@ -0,0 +1,6 @@
+## To train models
+
+`python run_models.py --config configs/config_ccle.yaml`
+
+This command will the model using four different drug encoders for 10 different data splits.
+Will add more details..
diff --git a/model4/configs/config_ccle.yaml b/model4/configs/config_ccle.yaml
@@ -0,0 +1,5 @@
+epochs: 100
+data_type: CCLE
+metric: auc
+out_dir: cmp_ccle
+cuda_id: 1
diff --git a/model4/configs/config_ccle_gexp.yaml b/model4/configs/config_ccle_gexp.yaml
@@ -0,0 +1,6 @@
+epochs: 100
+data_type: CCLE
+metric: auc
+out_dir: ccle_gexp
+cuda_id: 4
+use_proteomics_data: 0 # 0: use gene expression, 1: use proteomics
diff --git a/model4/configs/config_ccle_prot.yaml b/model4/configs/config_ccle_prot.yaml
@@ -0,0 +1,5 @@
+epochs: 100
+data_type: CCLE
+metric: auc
+out_dir: ccle_prot
+cuda_id: 2
diff --git a/model4/configs/config_ctrpv2.yaml b/model4/configs/config_ctrpv2.yaml
@@ -0,0 +1,5 @@
+epochs: 100
+data_type: CTRPv2
+metric: auc
+out_dir: cmp_ctrpv2
+cuda_id: 4
diff --git a/model4/configs/config_ctrpv2_gexp.yaml b/model4/configs/config_ctrpv2_gexp.yaml
@@ -0,0 +1,6 @@
+epochs: 100
+data_type: CTRPv2
+metric: auc
+out_dir: ctrpv2_gexp
+cuda_id: 4
+use_proteomics_data: 0 # 0: use gene expression, 1: use proteomics
diff --git a/model4/configs/config_ctrpv2_prot.yaml b/model4/configs/config_ctrpv2_prot.yaml
@@ -0,0 +1,5 @@
+epochs: 100
+data_type: CTRPv2
+metric: auc
+out_dir: ctrpv2_prot
+cuda_id: 3