def post_enhc(iter_index, json_file):
iter_name = make_iter_name(iter_index)
work_path = iter_name + "/" + enhc_name + "/"
fp = open(json_file, 'r')
jdata = json.load(fp)
gmx_split = jdata["gmx_split_traj"]
gmx_split_log = "gmx_split.log"
gmx_split_cmd = cmd_append_log(gmx_split, gmx_split_log)
all_task = glob.glob(work_path + "/[0-9]*[0-9]")
all_task.sort()
cwd = os.getcwd()
numb_walkers = jdata["numb_walkers"]
for ii in range(numb_walkers):
walker_path = work_path + make_walker_name(ii) + "/"
os.chdir(walker_path)
if os.path.isdir("confs"):
shutil.rmtree("confs")
os.makedirs("confs")
os.chdir(cwd)
global exec_machine
exec_hosts(MachineLocal, gmx_split_cmd, 1, all_task, None)
for ii in range(numb_walkers):
walker_path = work_path + make_walker_name(ii) + "/"
angles = np.loadtxt(walker_path + enhc_out_plm)
np.savetxt(walker_path + enhc_out_angle, angles[:, 1:], fmt="%.6f")
def post_sits_iter(sits_iter_index, json_file):
sits_dir = join("sits", make_iter_name(sits_iter_index))
fp = open(json_file, 'r')
jdata = json.load(fp)
sits_param = jdata.get("sits_settings", None)
tempf = np.linspace(sits_param["sits-t-low"], sits_param["sits-t-high"],
sits_param["sits-t-numbers"])
CONSTANT_kB = 0.008314472
beta_k = 1.0 / (CONSTANT_kB * tempf)
beta_0 = 1.0 / (CONSTANT_kB * sits_param["sits-t-ref"])
if not os.path.exists(join(sits_dir, "beta_k.dat")):
np.savetxt(join(sits_dir, "beta_k.dat"), beta_k)
if not os.path.exists(join("sits", "beta_0.dat")):
np.savetxt(join("sits", "beta_0.dat"), np.array([beta_0]))
walker_dir = join(sits_dir, enhc_name, make_walker_name(0))
cmd_save_sits = ""
cmd_save_sits += "tail -n 1 %s > %s\n" % (join(
walker_dir, "sits_nk.dat"), join(sits_dir, "log_nk.dat"))
cmd_save_sits += "tail -n 1 %s > %s\n" % (join(
walker_dir, "sits_norm.dat"), join(sits_dir, "log_norm.dat"))
cmd_save_sits += "cp %s %s\n" % (join(
sits_dir, "log_nk.dat"), join("sits", "log_nk.dat"))
cmd_save_sits += "cp %s %s\n" % (join(
sits_dir, "log_norm.dat"), join("sits", "log_norm.dat"))
cmd_save_sits += "cp %s %s\n" % (join(
sits_dir, "beta_k.dat"), join("sits", "beta_k.dat"))
sp.check_call(cmd_save_sits, shell=True)
def make_sits_iter(sits_iter_index, json_file, graph_files):
make_enhc(sits_iter_index, json_file, graph_files, sits_iter=True)
if sits_iter_index > 0:
old_dir = join("sits", make_iter_name(sits_iter_index - 1))
walker_dir = join("sits", make_iter_name(sits_iter_index), enhc_name,
make_walker_name(0))
try:
shutil.copyfile(join(old_dir, "log_nk.dat"),
join(walker_dir, "log_nk.dat"))
shutil.copyfile(join(old_dir, "log_norm.dat"),
join(walker_dir, "log_norm.dat"))
except:
pass
def run_enhc(iter_index, json_file):
iter_name = make_iter_name(iter_index)
work_path = iter_name + "/" + enhc_name + "/"
fp = open(json_file, 'r')
jdata = json.load(fp)
gmx_prep = jdata["gmx_prep"]
gmx_run = jdata["gmx_run"]
enhc_thread = jdata["bias_thread"]
gmx_run = gmx_run + (" -nt %d " % enhc_thread)
gmx_prep_log = "gmx_grompp.log"
gmx_run_log = "gmx_mdrun.log"
# assuming at least one walker
graph_files = glob.glob(work_path + (make_walker_name(0)) + "/*.pb")
if len(graph_files) != 0:
gmx_run = gmx_run + " -plumed " + enhc_plm
else:
gmx_run = gmx_run + " -plumed " + enhc_bf_plm
gmx_prep_cmd = cmd_append_log(gmx_prep, gmx_prep_log)
gmx_run_cmd = cmd_append_log(gmx_run, gmx_run_log)
numb_walkers = jdata["numb_walkers"]
batch_jobs = jdata['batch_jobs']
batch_time_limit = jdata['batch_time_limit']
batch_modules = jdata['batch_modules']
batch_sources = jdata['batch_sources']
all_task = glob.glob(work_path + "/[0-9]*[0-9]")
all_task.sort()
global exec_machine
exec_hosts(MachineLocal, gmx_prep_cmd, 1, all_task, None)
if batch_jobs:
exec_batch(gmx_run_cmd,
enhc_thread,
1,
all_task,
task_args=None,
time_limit=batch_time_limit,
modules=batch_modules,
sources=batch_sources)
else:
if len(all_task) == 1:
exec_hosts(MachineLocal, gmx_run_cmd, enhc_thread, all_task, None)
else:
exec_hosts_batch(exec_machine, gmx_run_cmd, enhc_thread, all_task,
None)
def make_temp(iter_index, json_file, graph_files):
graph_files.sort()
fp = open(json_file, 'r')
jdata = json.load(fp)
numb_walkers = jdata["numb_walkers"]
template_dir = jdata["template_dir"]
nsteps = jdata["temp_nsteps"]
frame_freq = jdata["temp_frame_freq"]
start_temp = jdata["start_temp"]
iter_name = make_iter_name(iter_index)
work_path = iter_name + "/" + temp_name + "/"
mol_path = template_dir + "/" + mol_name + "/"
temp_path = template_dir + "/" + temp_name + "/"
conf_list = glob.glob(mol_path + "conf*gro")
conf_list.sort()
assert (len(conf_list) >=
numb_walkers), "not enough conf files in mol dir %s" % mol_path
create_path(work_path)
for walker_idx in range(numb_walkers):
walker_path = work_path + make_walker_name(walker_idx) + "/"
create_path(walker_path)
# copy md ifles
for ii in mol_files:
if os.path.exists(walker_path + ii):
os.remove(walker_path + ii)
shutil.copy(mol_path + ii, walker_path)
# copy conf file
conf_file = conf_list[walker_idx]
if os.path.exists(walker_path + "conf.gro"):
os.remove(walker_path + "conf.gro")
shutil.copy(conf_file, walker_path + "conf.gro")
# if have prev confout.gro, use as init conf
if (iter_index > 0):
prev_temp_path = make_iter_name(
iter_index -
1) + "/" + temp_name + "/" + make_walker_name(walker_idx) + "/"
prev_temp_path = os.path.abspath(prev_temp_path) + "/"
if os.path.isfile(prev_temp_path + "confout.gro"):
os.remove(walker_path + "conf.gro")
os.symlink(prev_temp_path + "confout.gro",
walker_path + "conf.gro")
log_task("use conf of iter " + make_iter_name(iter_index - 1) +
" walker " + make_walker_name(walker_idx))
# copy temp file
for ii in temp_files:
if os.path.exists(walker_path + ii):
os.remove(walker_path + ii)
shutil.copy(temp_path + ii, walker_path)
# copy graph files
for ii in graph_files:
file_name = os.path.basename(ii)
abs_path = os.path.abspath(ii)
if os.path.exists(walker_path + file_name):
os.remove(walker_path + file_name)
os.symlink(abs_path, walker_path + file_name)
# config MD
mol_conf_file = walker_path + "grompp.mdp"
make_grompp_enhc(mol_conf_file, nsteps, frame_freq)
# config plumed
if iter_index == 0:
cur_temp = start_temp
else:
cur_temp = np.loadtxt(os.path.join(prev_temp_path, "next.temp"))
log_task(("use temp of %f") % (cur_temp))
log_task(("length of traj %d") % (nsteps))
np.savetxt(os.path.join(walker_path, 'cur.temp'), [cur_temp])
plm_conf = walker_path + temp_plm
replace(plm_conf, "TEMP=[^ ]* ", ("TEMP=%s " % cur_temp))
replace(plm_conf, "STRIDE=[^ ]* ", ("STRIDE=%d " % frame_freq))
replace(plm_conf, "FILE=[^ ]* ", ("FILE=%s " % temp_out_plm))
def post_temp(iter_index, json_file):
iter_name = make_iter_name(iter_index)
work_path = iter_name + "/" + temp_name + "/"
fp = open(json_file, 'r')
jdata = json.load(fp)
gmx_split = jdata["gmx_split_traj"]
gmx_split_log = "gmx_split.log"
gmx_split_cmd = cmd_append_log(gmx_split, gmx_split_log)
temp_test_intvl = jdata['temp_test_intvl']
dev_trust_lvl = jdata['dev_trust_lvl']
temp_incr_ratio = jdata['temp_incr_ratio']
temp_decr_ratio = jdata['temp_decr_ratio']
temp_incr_lvl = jdata['temp_incr_lvl']
temp_decr_lvl = jdata['temp_decr_lvl']
phys_temp = jdata['phys_temp']
max_temp = jdata['max_temp']
all_task = glob.glob(work_path + "/[0-9]*[0-9]")
all_task.sort()
cwd = os.getcwd()
numb_walkers = jdata["numb_walkers"]
for ii in range(numb_walkers):
walker_path = work_path + make_walker_name(ii) + "/"
os.chdir(walker_path)
if os.path.isdir("confs"):
shutil.rmtree("confs")
os.makedirs("confs")
os.chdir(cwd)
global exec_machine
exec_hosts(MachineLocal, gmx_split_cmd, 1, all_task, None)
walker_stop_flags = []
for ii in range(numb_walkers):
walker_path = work_path + make_walker_name(ii) + "/"
angles = np.loadtxt(walker_path + temp_out_plm)
angles = angles[:, 1:]
np.savetxt(walker_path + temp_out_angle, angles, fmt="%.6f")
graph_files = glob.glob(walker_path + "/*.pb")
cur_temp = np.loadtxt(os.path.join(walker_path, 'cur.temp'))
if len(graph_files) != 0:
next_temp, dec_list, model_dev\
= make_next_temp(graph_files,
cur_temp,
angles,
temp_test_intvl,
dev_trust_lvl,
temp_incr_ratio,
temp_decr_ratio,
temp_incr_lvl,
temp_decr_lvl,
phys_temp,
max_temp)
model_dev = np.concatenate(
(angles, np.reshape(model_dev, [-1, 1])), axis=1)
np.savetxt(os.path.join(walker_path, 'model.devi'), model_dev)
np.savetxt(os.path.join(walker_path, 'dec_list'), dec_list)
else:
next_temp = cur_temp
np.savetxt(os.path.join(walker_path, 'next.temp'), [next_temp])
if cur_temp == max_temp and next_temp == max_temp:
walker_stop_flags.append(True)
else:
walker_stop_flags.append(False)
# return if the iteration continues
if all(walker_stop_flags):
return False
else:
return True
def run_enhc(iter_index, json_file):
iter_name = make_iter_name(iter_index)
work_path = iter_name + "/" + enhc_name + "/"
fp = open(json_file, 'r')
jdata = json.load(fp)
bPosre = jdata.get("gmx_posre", False)
gmx_prep = jdata["gmx_prep"]
if bPosre:
gmx_prep += " -f grompp_restraint.mdp -r conf_init.gro"
gmx_run = jdata["gmx_run"]
enhc_thread = jdata["bias_thread"]
gmx_run = gmx_run + (" -nt %d " % enhc_thread)
gmx_prep_log = "gmx_grompp.log"
gmx_run_log = "gmx_mdrun.log"
# assuming at least one walker
graph_files = glob.glob(work_path + (make_walker_name(0)) + "/*.pb")
if len(graph_files) != 0:
gmx_run = gmx_run + " -plumed " + enhc_plm
else:
gmx_run = gmx_run + " -plumed " + enhc_bf_plm
gmx_prep_cmd = cmd_append_log(gmx_prep, gmx_prep_log)
gmx_run_cmd = cmd_append_log(gmx_run, gmx_run_log)
numb_walkers = jdata["numb_walkers"]
batch_jobs = jdata['batch_jobs']
batch_time_limit = jdata['batch_time_limit']
batch_modules = jdata['batch_modules']
batch_sources = jdata['batch_sources']
# print('debug', glob.glob(work_path + "/[0-9]*[0-9]"))
# all_task = glob.glob(work_path + "/[0-9]*[0-9]")
all_task = list(
filter(lambda x: os.path.isdir(x),
glob.glob(work_path + "/[0-9]*[0-9]")))
all_task.sort()
# all_task_basedir = [os.path.relpath(ii, work_path) for ii in all_task]
# print('run_enhc:work_path', work_path)
# print('run_enhc:gmx_prep_cmd:', gmx_prep_cmd)
# print('run_enhc:gmx_run_cmd:', gmx_run_cmd)
# print('run_enhc:all_task:', all_task)
# print('run_enhc:all_task_basedir:', all_task_basedir)
# print('run_enhc:batch_jobs:', batch_jobs)
#
# lazy_local_context = LazyLocalContext(local_root='./', work_profile=None)
# # pbs = PBS(context=lazy_local_context)
# slurm = Slurm(context=lazy_local_context)
# gmx_prep_task = [Task(command=gmx_prep_cmd, task_work_path=ii, outlog='gmx_grompp.log', errlog='gmx_grompp.err') for
# ii in all_task_basedir]
# gmx_prep_submission = Submission(work_base=work_path, resources=resources, batch=slurm, task_list=gmx_prep_task)
#
# gmx_prep_submission.run_submission()
#
# gmx_run_task = [Task(command=gmx_run_cmd, task_work_path=ii, outlog='gmx_mdrun.log', errlog='gmx_mdrun.log') for ii
# in all_task_basedir]
# gmx_run_submission = Submission(work_base=work_path, resources=resources, batch=slurm, task_list=gmx_run_task)
# gmx_run_submission.run_submission()
global exec_machine
exec_hosts(MachineLocal, gmx_prep_cmd, 1, all_task, None)
if batch_jobs:
exec_batch(gmx_run_cmd,
enhc_thread,
1,
all_task,
task_args=None,
time_limit=batch_time_limit,
modules=batch_modules,
sources=batch_sources)
else:
if len(all_task) == 1:
exec_hosts(MachineLocal, gmx_run_cmd, enhc_thread, all_task, None)
else:
exec_hosts_batch(exec_machine, gmx_run_cmd, enhc_thread, all_task,
None)
def make_enhc(iter_index, json_file, graph_files):
graph_files.sort()
fp = open(json_file, 'r')
jdata = json.load(fp)
bPosre = jdata.get("gmx_posre", False)
numb_walkers = jdata["numb_walkers"]
template_dir = jdata["template_dir"]
enhc_trust_lvl_1 = jdata["bias_trust_lvl_1"]
enhc_trust_lvl_2 = jdata["bias_trust_lvl_2"]
nsteps = jdata["bias_nsteps"]
frame_freq = jdata["bias_frame_freq"]
num_of_cluster_threshhold = jdata["num_of_cluster_threshhold"]
iter_name = make_iter_name(iter_index)
work_path = iter_name + "/" + enhc_name + "/"
mol_path = template_dir + "/" + mol_name + "/"
enhc_path = template_dir + "/" + enhc_name + "/"
conf_list = glob.glob(mol_path + "conf*gro")
conf_list.sort()
assert (len(conf_list) >=
numb_walkers), "not enough conf files in mol dir %s" % mol_path
create_path(work_path)
kappa = np.linspace(2, 16, 8)
dis_kappa = np.linspace(4.5, 1, 8) # a weak distance restraint.
for walker_idx in range(numb_walkers):
kk = kappa[walker_idx]
walker_path = work_path + make_walker_name(walker_idx) + "/"
create_path(walker_path)
# copy md ifles
copy_file_list(mol_files, mol_path, walker_path)
# copy conf file
conf_file = conf_list[walker_idx]
if os.path.exists(walker_path + "conf.gro"):
os.remove(walker_path + "conf.gro")
try:
shutil.copy(conf_file, walker_path + "conf.gro")
except:
pass
if os.path.exists(walker_path + "conf_init.gro"):
os.remove(walker_path + "conf_init.gro")
shutil.copy(conf_file, walker_path + "conf_init.gro")
# if have prev confout.gro, use as init conf
if (iter_index > 0):
kk = kappa[(walker_idx + iter_index) % 8]
prev_enhc_path = make_iter_name(
iter_index -
1) + "/" + enhc_name + "/" + make_walker_name(walker_idx) + "/"
prev_enhc_path = os.path.abspath(prev_enhc_path) + "/"
if os.path.isfile(prev_enhc_path + "confout.gro"):
os.remove(walker_path + "conf.gro")
os.remove(walker_path + "conf_init.gro")
os.symlink(prev_enhc_path + "confout.gro",
walker_path + "conf.gro")
os.symlink(prev_enhc_path + "conf_init.gro",
walker_path + "conf_init.gro")
log_task("use conf of iter " + make_iter_name(iter_index - 1) +
" walker " + make_walker_name(walker_idx))
else:
raise RuntimeError("cannot find prev output conf file " +
prev_enhc_path + 'confout.gro')
log_task("use conf of iter " + make_iter_name(iter_index - 1) +
" walker " + make_walker_name(walker_idx))
###########################################
num_of_cluster = np.loadtxt(prev_enhc_path + 'num_of_cluster.dat')
pre_trust_lvl1 = np.loadtxt(prev_enhc_path + 'trust_lvl1.dat')
if num_of_cluster < num_of_cluster_threshhold:
enhc_trust_lvl_1 = pre_trust_lvl1 * 1.5
enhc_trust_lvl_2 = enhc_trust_lvl_1 + 1
else:
enhc_trust_lvl_1 = jdata["bias_trust_lvl_1"]
enhc_trust_lvl_2 = enhc_trust_lvl_1 + 1
if enhc_trust_lvl_1 > jdata["bias_trust_lvl_1"] * 8:
enhc_trust_lvl_1 = jdata["bias_trust_lvl_1"]
enhc_trust_lvl_2 = enhc_trust_lvl_1 + 1
np.savetxt(walker_path + 'trust_lvl1.dat', [enhc_trust_lvl_1],
fmt='%.6f')
# copy enhc file
for ii in enhc_files:
if os.path.exists(walker_path + ii):
os.remove(walker_path + ii)
try:
shutil.copy(enhc_path + ii, walker_path)
except:
pass
# copy graph files
for ii in graph_files:
file_name = os.path.basename(ii)
abs_path = os.path.abspath(ii)
if os.path.exists(walker_path + file_name):
os.remove(walker_path + file_name)
os.symlink(abs_path, walker_path + file_name)
# config MD
mol_conf_file = walker_path + "grompp.mdp"
if bPosre:
mol_conf_file = walker_path + "grompp_restraint.mdp"
make_grompp_enhc(mol_conf_file, nsteps, frame_freq)
# config plumed
graph_list = ""
counter = 0
for ii in graph_files:
file_name = os.path.basename(ii)
if counter == 0:
graph_list = "%s" % file_name
else:
graph_list = "%s,%s" % (graph_list, file_name)
counter = counter + 1
posre_files = glob.glob(walker_path + 'posre*.itp')
for posre_file in posre_files:
replace(posre_file, 'TEMP', '%d' % kk)
plm_conf = walker_path + enhc_plm
replace(plm_conf, "MODEL=[^ ]* ", ("MODEL=%s " % graph_list))
replace(plm_conf, "TRUST_LVL_1=[^ ]* ",
("TRUST_LVL_1=%f " % enhc_trust_lvl_1))
replace(plm_conf, "TRUST_LVL_2=[^ ]* ",
("TRUST_LVL_2=%f " % enhc_trust_lvl_2))
replace(plm_conf, "STRIDE=[^ ]* ", ("STRIDE=%d " % frame_freq))
replace(plm_conf, "FILE=[^ ]* ", ("FILE=%s " % enhc_out_plm))
plm_bf_conf = walker_path + enhc_bf_plm
replace(plm_bf_conf, "STRIDE=[^ ]* ", ("STRIDE=%d " % frame_freq))
replace(plm_bf_conf, "FILE=[^ ]* ", ("FILE=%s " % enhc_out_plm))
# molecule_name = os.getcwd().split('/')[-1].split('.')[0]
# distance_list = get_distance('../{}/{}/{}.pdb'.format(molecule_name, molecule_name, molecule_name))
# CA_list = get_CA_atom(walker_path + 'conf.gro')
# ret = add_distance_restrain(CA_list, distance_list, dis_buttom=0.2, dis_kappa=dis_kappa[(walker_idx+iter_index)%8])
# with open(plm_conf, 'a') as plm:
# plm.write(ret)
# with open(plm_bf_conf, 'a') as plm:
# plm.write(ret)
if len(graph_list) == 0:
log_task("brute force MD without NN acc")
else:
log_task("use NN model(s): " + graph_list)
log_task("set trust l1 and l2: %f %f" %
(enhc_trust_lvl_1, enhc_trust_lvl_2))
def make_enhc(iter_index, json_file, graph_files):
graph_files.sort()
fp = open(json_file, 'r')
jdata = json.load(fp)
numb_walkers = jdata["numb_walkers"]
template_dir = jdata["template_dir"]
enhc_trust_lvl_1 = jdata["bias_trust_lvl_1"]
enhc_trust_lvl_2 = jdata["bias_trust_lvl_2"]
nsteps = jdata["bias_nsteps"]
frame_freq = jdata["bias_frame_freq"]
num_of_cluster_threshhold = jdata["num_of_cluster_threshhold"]
iter_name = make_iter_name(iter_index)
work_path = iter_name + "/" + enhc_name + "/"
mol_path = template_dir + "/" + mol_name + "/"
enhc_path = template_dir + "/" + enhc_name + "/"
conf_list = glob.glob(mol_path + "conf*gro")
conf_list.sort()
assert (len(conf_list) >=
numb_walkers), "not enough conf files in mol dir %s" % mol_path
create_path(work_path)
for walker_idx in range(numb_walkers):
walker_path = work_path + make_walker_name(walker_idx) + "/"
create_path(walker_path)
# copy md ifles
for ii in mol_files:
if os.path.exists(walker_path + ii):
os.remove(walker_path + ii)
shutil.copy(mol_path + ii, walker_path)
# copy conf file
conf_file = conf_list[walker_idx]
if os.path.exists(walker_path + "conf.gro"):
os.remove(walker_path + "conf.gro")
shutil.copy(conf_file, walker_path + "conf.gro")
# if have prev confout.gro, use as init conf
if (iter_index > 0):
prev_enhc_path = make_iter_name(
iter_index -
1) + "/" + enhc_name + "/" + make_walker_name(walker_idx) + "/"
prev_enhc_path = os.path.abspath(prev_enhc_path) + "/"
if os.path.isfile(prev_enhc_path + "confout.gro"):
os.remove(walker_path + "conf.gro")
os.symlink(prev_enhc_path + "confout.gro",
walker_path + "conf.gro")
else:
raise RuntimeError("cannot find prev output conf file " +
prev_enhc_path + 'confout.gro')
log_task("use conf of iter " + make_iter_name(iter_index - 1) +
" walker " + make_walker_name(walker_idx))
###########################################
num_of_cluster = np.loadtxt(prev_enhc_path + 'num_of_cluster.dat')
pre_trust_lvl1 = np.loadtxt(prev_enhc_path + 'trust_lvl1.dat')
if num_of_cluster < num_of_cluster_threshhold:
enhc_trust_lvl_1 = pre_trust_lvl1 * 1.5
enhc_trust_lvl_2 = enhc_trust_lvl_1 + 1
else:
enhc_trust_lvl_1 = jdata["bias_trust_lvl_1"]
enhc_trust_lvl_2 = enhc_trust_lvl_1 + 1
if enhc_trust_lvl_1 > jdata["bias_trust_lvl_1"] * 8:
enhc_trust_lvl_1 = jdata["bias_trust_lvl_1"]
enhc_trust_lvl_2 = enhc_trust_lvl_1 + 1
np.savetxt(walker_path + 'trust_lvl1.dat', [enhc_trust_lvl_1],
fmt='%.6f')
# copy enhc file
for ii in enhc_files:
if os.path.exists(walker_path + ii):
os.remove(walker_path + ii)
shutil.copy(enhc_path + ii, walker_path)
# copy graph files
for ii in graph_files:
file_name = os.path.basename(ii)
abs_path = os.path.abspath(ii)
if os.path.exists(walker_path + file_name):
os.remove(walker_path + file_name)
os.symlink(abs_path, walker_path + file_name)
# config MD
mol_conf_file = walker_path + "grompp.mdp"
make_grompp_enhc(mol_conf_file, nsteps, frame_freq)
# config plumed
graph_list = ""
counter = 0
for ii in graph_files:
file_name = os.path.basename(ii)
if counter == 0:
graph_list = "%s" % file_name
else:
graph_list = "%s,%s" % (graph_list, file_name)
counter = counter + 1
plm_conf = walker_path + enhc_plm
replace(plm_conf, "MODEL=[^ ]* ", ("MODEL=%s " % graph_list))
replace(plm_conf, "TRUST_LVL_1=[^ ]* ",
("TRUST_LVL_1=%f " % enhc_trust_lvl_1))
replace(plm_conf, "TRUST_LVL_2=[^ ]* ",
("TRUST_LVL_2=%f " % enhc_trust_lvl_2))
replace(plm_conf, "STRIDE=[^ ]* ", ("STRIDE=%d " % frame_freq))
replace(plm_conf, "FILE=[^ ]* ", ("FILE=%s " % enhc_out_plm))
plm_bf_conf = walker_path + enhc_bf_plm
replace(plm_bf_conf, "STRIDE=[^ ]* ", ("STRIDE=%d " % frame_freq))
replace(plm_bf_conf, "FILE=[^ ]* ", ("FILE=%s " % enhc_out_plm))
if len(graph_list) == 0:
log_task("brute force MD without NN acc")
else:
log_task("use NN model(s): " + graph_list)
log_task("set trust l1 and l2: %f %f" %
(enhc_trust_lvl_1, enhc_trust_lvl_2))
