def create_out_dir(params):
"""Setup the output directory."""
params.in_tr_data_dir = utils.clean_last_slash_if_any(
params.in_tr_data_dir)
params.in_val_data_dir = utils.clean_last_slash_if_any(
params.in_val_data_dir)
params.ood_val_data_dir = utils.clean_last_slash_if_any(
params.ood_val_data_dir)
sub_dir = ('generative_l%d_bs%d_lr%.4f'
'_hr%d_nr%s_reg%s_regw%.6f_fi%d_mt%.2f') % (
params.seq_len, params.batch_size, params.learning_rate,
params.hidden_lstm_size, params.norm_lstm, params.reg_type,
params.reg_weight, params.filter_label,
params.mutation_rate)
log_dir = os.path.join(params.out_dir, sub_dir, 'log')
params.add_hparam('log_dir_in_tr', os.path.join(log_dir, 'in_tr'))
params.add_hparam('log_dir_in_val', os.path.join(log_dir, 'in_val'))
params.add_hparam('model_dir', log_dir.replace('log', 'model'))
if not tf.gfile.Exists(params.out_dir):
tf.gfile.MakeDirs(params.out_dir)
if not tf.gfile.Exists(params.log_dir_in_tr):
tf.gfile.MakeDirs(params.log_dir_in_tr)
if not tf.gfile.Exists(params.log_dir_in_val):
tf.gfile.MakeDirs(params.log_dir_in_val)
if not tf.gfile.Exists(params.model_dir):
tf.gfile.MakeDirs(params.model_dir)
tf.logging.info('model_dir=%s', params.model_dir)
def create_out_dir(params):
"""Setup the output directory."""
params.in_tr_data_dir = utils.clean_last_slash_if_any(
params.in_tr_data_dir)
params.in_val_data_dir = utils.clean_last_slash_if_any(
params.in_val_data_dir)
params.ood_val_data_dir = utils.clean_last_slash_if_any(
params.ood_val_data_dir)
sub_dir = ('%s_emb%s_l%d_bs%d_lr%.4f_nm%d_lm%d_hd%d'
'_t%.2f_l2r%.6f_rw%s_mr%.2f') % (
os.path.basename(params.in_tr_data_dir), params.embedding,
params.seq_len, params.batch_size, params.learning_rate,
params.num_motifs, params.len_motifs,
params.hidden_dense_size, params.temperature, params.l2_reg,
params.reweight_sample, params.mutation_rate)
log_dir = os.path.join(params.out_dir, sub_dir, 'log')
params.add_hparam('log_dir_in_tr', os.path.join(log_dir, 'tr'))
params.add_hparam('log_dir_in_val', os.path.join(log_dir, 'val'))
params.add_hparam('model_dir', log_dir.replace('log', 'model'))
if not tf.gfile.Exists(params.out_dir):
tf.gfile.MakeDirs(params.out_dir)
if not tf.gfile.Exists(params.log_dir_in_tr):
tf.gfile.MakeDirs(params.log_dir_in_tr)
if not tf.gfile.Exists(params.log_dir_in_val):
tf.gfile.MakeDirs(params.log_dir_in_val)
if not tf.gfile.Exists(params.model_dir):
tf.gfile.MakeDirs(params.model_dir)
tf.logging.info('model_dir=%s', params.model_dir)
