luost26 commited on
Commit
7efb4f9
1 Parent(s): 20002df
configs/test/abopt_singlecdr.yml ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ mode: abopt
2
+ model:
3
+ checkpoint: ../trained_models/codesign_single.pt
4
+
5
+ sampling:
6
+ seed: 2022
7
+ sample_structure: true
8
+ sample_sequence: true
9
+ cdrs:
10
+ - H_CDR3
11
+ num_samples: 100
12
+ optimize_steps:
13
+ - 1
14
+ - 2
15
+ - 4
16
+ - 8
17
+ - 16
18
+ - 32
19
+ - 64
20
+
21
+ dataset:
22
+ test:
23
+ type: sabdab
24
+ summary_path: ./data/sabdab_summary_all.tsv
25
+ chothia_dir: ./data/all_structures/chothia
26
+ processed_dir: ./data/processed
27
+ split: test
configs/test/codesign_multicdrs.yml ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ mode: multiple_cdrs
2
+ model:
3
+ checkpoint: ../trained_models/codesign_multicdrs.pt
4
+
5
+ sampling:
6
+ seed: 2022
7
+ sample_structure: true
8
+ sample_sequence: true
9
+ cdrs:
10
+ - H_CDR1
11
+ - H_CDR2
12
+ - H_CDR3
13
+ - L_CDR1
14
+ - L_CDR2
15
+ - L_CDR3
16
+ num_samples: 100
17
+
18
+ dataset:
19
+ test:
20
+ type: sabdab
21
+ summary_path: ./data/sabdab_summary_all.tsv
22
+ chothia_dir: ./data/all_structures/chothia
23
+ processed_dir: ./data/processed
24
+ split: test
configs/test/codesign_single.yml ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ mode: single_cdr
2
+ model:
3
+ checkpoint: ../trained_models/codesign_single.pt
4
+
5
+ sampling:
6
+ seed: 2022
7
+ sample_structure: true
8
+ sample_sequence: true
9
+ cdrs:
10
+ - H_CDR1
11
+ - H_CDR2
12
+ - H_CDR3
13
+ - L_CDR1
14
+ - L_CDR2
15
+ - L_CDR3
16
+ num_samples: 100
17
+
18
+ dataset:
19
+ test:
20
+ type: sabdab
21
+ summary_path: ./data/sabdab_summary_all.tsv
22
+ chothia_dir: ./data/all_structures/chothia
23
+ processed_dir: ./data/processed
24
+ split: test
configs/test/fixbb.yml ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ mode: single_cdr
2
+ model:
3
+ checkpoint: ../trained_models/fixbb.pt
4
+
5
+ sampling:
6
+ seed: 2022
7
+ sample_structure: false
8
+ sample_sequence: true
9
+ cdrs:
10
+ - H_CDR1
11
+ - H_CDR2
12
+ - H_CDR3
13
+ - L_CDR1
14
+ - L_CDR2
15
+ - L_CDR3
16
+ num_samples: 100
17
+
18
+ dataset:
19
+ test:
20
+ type: sabdab
21
+ summary_path: ./data/sabdab_summary_all.tsv
22
+ chothia_dir: ./data/all_structures/chothia
23
+ processed_dir: ./data/processed
24
+ split: test
data/examples/3QHF_Fv.pdb ADDED
The diff for this file is too large to render. See raw diff
 
data/examples/7DK2_AB_C.pdb ADDED
The diff for this file is too large to render. See raw diff
 
data/examples/Omicron_RBD.pdb ADDED
The diff for this file is too large to render. See raw diff