forked from amirbarati/conformation
-
Notifications
You must be signed in to change notification settings - Fork 0
/
b2ar_gprot_tica_config.py
59 lines (49 loc) · 1.21 KB
/
b2ar_gprot_tica_config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
import os
import sys
import csv
exacycle = False
n_clusters = 100
lag_time = 5
if exacycle:
lag_time *= 2
msm_lag_time = 5
n_components = 25
n_samples = 10
n_macrostates = 25
n_trees = 100
precision = "SP"
sparse = True
wolf = True
is_sparse = ""
if(sparse): is_sparse = "sparse-"
shrinkage_string = ""
rho_string = ""
if wolf and not sparse:
wolf_string = "_wolf_"
shrinkage = None
shrinkage_string = "autoShrinkage"
rho = None
rho_string = ""
elif wolf and sparse:
wolf_string = "_wolf_"
shrinkage = None
shrinkage_string = "autoShrinkage"
rho = 0.05
rho_string = "_rho0pt05"
else:
wolf_string = ""
shrinkage = 0.001
shrinkage_string = "0pt001"
rho = None
rho_string = ""
traj_ext = ".h5"
base = "/home/enf/md_simulations/b2ar/2RH1-1azs"
traj_dir = "/home/enf/md_simulations/b2ar/2RH1-1azs/all_reimaged"
inactive_dir = "/home/enf/md_simulations/b2ar/2RH1-1azs/2RH1_prepped.pdb"
active_dir = "/home/enf/md_simulations/b2ar/2RH1-1azs/3P0G_pymol_prepped.pdb"
pnas_features_dir = "/home/enf/md_simulations/b2ar/2RH1-1azs/pnas_features"
if not os.path.exists(pnas_features_dir): os.makedirs(pnas_features_dir)
iterative = False
featurize_parallel = True
sampling_method = "random"
precision = "SP"