def main(adjustments_path):
# read adjustments
print('reading adjustments...', flush=True)
designpath_selectedstep = {}
with open(adjustments_path, mode='rt', encoding='utf-8', errors='surrogateescape') as fr:
for line in fr:
design_path, selected_step = [x.strip() for x in line.split('\t')]
designpath_selectedstep[design_path] = int(selected_step)
print('found {0!s} adjustments...'.format(len(designpath_selectedstep)), flush=True)
# make adjustments
print('making adjustments...', flush=True)
for didx, (design_path, selected_step) in enumerate(designpath_selectedstep.items()):
print('working on {0}...'.format(design_path), flush=True)
print('selected step:{0!s}...'.format(selected_step), flush=True)
# load design
print('loading design...', flush=True)
with open(design_path, mode='rt', encoding='utf-8', errors='surrogateescape') as fr:
d = json.load(fr)
if 'apply_activation_to_embedding' not in d: # for legacy code
d['apply_activation_to_embedding'] = True
if 'use_batchnorm' not in d: # for legacy code
d['use_batchnorm'] = False
if 'skip_layerwise_training' not in d: # for legacy code
d['skip_layerwise_training'] = False
phase = d['training_schedule'][-1]
d['current_hidden_layer'] = phase['hidden_layer']
d['current_finetuning_run'] = phase['finetuning_run']
d['current_epochs'] = phase['epochs']
# load data
if didx == 0:
print('loading data...', flush=True)
partitions = ['train', 'valid', 'test']
dataset = {}
for partition in partitions:
if partition == 'train':
dataset[partition] = datasetIO.load_datamatrix('{0}/{1}.pickle'.format(d['input_path'], 'valid'))
dataset[partition].append(datasetIO.load_datamatrix('{0}/{1}.pickle'.format(d['input_path'], 'test')), 0)
elif partition == 'valid':
dataset[partition] = datasetIO.load_datamatrix('{0}/{1}.pickle'.format(d['input_path'], 'train'))
else:
dataset[partition] = datasetIO.load_datamatrix('{0}/{1}.pickle'.format(d['input_path'], partition))
# finish configuration
print('finishing configuration...', flush=True)
# specify activation function
if d['activation_function'] == 'tanh':
activation_function = {'np':sdae_apply_functions.tanh}
elif d['activation_function'] == 'relu':
activation_function = {'np':sdae_apply_functions.relu}
elif d['activation_function'] == 'elu':
activation_function = {'np':sdae_apply_functions.elu}
elif d['activation_function'] == 'sigmoid':
activation_function = {'np':sdae_apply_functions.sigmoid}
# initialize model architecture (number of layers and dimension of each layer)
d['current_dimensions'] = d['all_dimensions'][:d['current_hidden_layer']+1] # dimensions of model up to current depth
# specify embedding function for current training phase
# we want the option of skipping the embedding activation function to apply only to the full model
if not d['apply_activation_to_embedding'] and d['current_dimensions'] == d['all_dimensions']:
d['current_apply_activation_to_embedding'] = False
else:
d['current_apply_activation_to_embedding'] = True
# specify rows and columns of figure showing data reconstructions
d['reconstruction_rows'] = int(np.round(np.sqrt(np.min([100, dataset['valid'].shape[0]])/2)))
d['reconstruction_cols'] = 2*d['reconstruction_rows']
# move files
print('moving files...', flush=True)
if os.path.exists('{0}/intermediate_variables_layer{1!s}_finetuning{2!s}_step{3!s}.pickle'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run'], selected_step)):
if os.path.exists('{0}/variables_layer{1!s}_finetuning{2!s}.pickle'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run'])):
shutil.move('{0}/variables_layer{1!s}_finetuning{2!s}.pickle'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run']),
'{0}/variables_layer{1!s}_finetuning{2!s}_old.pickle'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run']))
shutil.copyfile('{0}/intermediate_variables_layer{1!s}_finetuning{2!s}_step{3!s}.pickle'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run'], selected_step),
'{0}/variables_layer{1!s}_finetuning{2!s}.pickle'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run']))
else:
print('variables do no exist for selected step! skipping...', flush=True)
continue
if d['use_batchnorm']:
if os.path.exists('{0}/intermediate_batchnorm_variables_layer{1!s}_finetuning{2!s}_step{3!s}.pickle'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run'], selected_step)):
if os.path.exists('{0}/batchnorm_variables_layer{1!s}_finetuning{2!s}.pickle'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run'])):
shutil.move('{0}/batchnorm_variables_layer{1!s}_finetuning{2!s}.pickle'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run']),
'{0}/batchnorm_variables_layer{1!s}_finetuning{2!s}_old.pickle'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run']))
shutil.copyfile('{0}/intermediate_batchnorm_variables_layer{1!s}_finetuning{2!s}_step{3!s}.pickle'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run'], selected_step),
'{0}/batchnorm_variables_layer{1!s}_finetuning{2!s}.pickle'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run']))
else:
print('batchnorm variables do no exist for selected step! skipping...', flush=True)
continue
# load model variables
print('loading model variables...', flush=True)
with open('{0}/variables_layer{1!s}_finetuning{2!s}.pickle'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run']), 'rb') as fr:
W, Be, Bd = pickle.load(fr)[1:] # global_step, W, bencode, bdecode
if d['use_batchnorm']:
with open('{0}/batchnorm_variables_layer{1!s}_finetuning{2!s}.pickle'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run']), 'rb') as fr:
batchnorm_variables = pickle.load(fr) # gammas, betas, moving_means, moving_variances
batchnorm_encode_variables, batchnorm_decode_variables = sdae_apply_functions.align_batchnorm_variables(batchnorm_variables, d['current_apply_activation_to_embedding'], d['apply_activation_to_output'])
# load reporting variables
print('loading reporting variables...', flush=True)
if os.path.exists('{0}/optimization_path_layer{1!s}_finetuning{2!s}.pickle'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run'])):
with open('{0}/optimization_path_layer{1!s}_finetuning{2!s}.pickle'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run']), 'rb') as fr:
optimization_path = pickle.load(fr)
reporting_steps = optimization_path['reporting_steps']
valid_losses = optimization_path['valid_losses']
train_losses = optimization_path['train_losses']
valid_noisy_losses = optimization_path['valid_noisy_losses']
train_noisy_losses = optimization_path['train_noisy_losses']
else:
reporting_steps = np.zeros(0, dtype='int32')
valid_losses = np.zeros(0, dtype='float32')
train_losses = np.zeros(0, dtype='float32')
valid_noisy_losses = np.zeros(0, dtype='float32')
train_noisy_losses = np.zeros(0, dtype='float32')
with open('{0}/log_layer{1!s}_finetuning{2!s}.txt'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run']), 'rt') as fr:
fr.readline()
for line in fr:
step, train_loss, valid_loss, train_noisy_loss, valid_noisy_loss, time = [float(x.strip()) for x in line.split('\t')]
reporting_steps = np.insert(reporting_steps, reporting_steps.size, step)
valid_losses = np.insert(valid_losses, valid_losses.size, valid_loss)
train_losses = np.insert(train_losses, train_losses.size, train_loss)
valid_noisy_losses = np.insert(valid_noisy_losses, valid_noisy_losses.size, valid_noisy_loss)
train_noisy_losses = np.insert(train_noisy_losses, train_noisy_losses.size, train_noisy_loss)
with open('{0}/optimization_path_layer{1!s}_finetuning{2!s}.pickle'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run']), 'wb') as fw:
pickle.dump({'reporting_steps':reporting_steps, 'valid_losses':valid_losses, 'train_losses':train_losses, 'valid_noisy_losses':valid_noisy_losses, 'train_noisy_losses':train_noisy_losses}, fw)
# compute embedding and reconstruction
print('computing embedding and reconstruction...', flush=True)
recon = {}
embed = {}
error = {}
embed_preactivation = {}
for partition in partitions:
if d['use_batchnorm']:
recon[partition], embed[partition], error[partition] = sdae_apply_functions.encode_and_decode(dataset[partition], W, Be, Bd, activation_function['np'], d['current_apply_activation_to_embedding'], d['apply_activation_to_output'], return_embedding=True, return_reconstruction_error=True, bn_encode_variables=batchnorm_encode_variables, bn_decode_variables=batchnorm_decode_variables)
embed_preactivation[partition] = sdae_apply_functions.encode(dataset[partition], W, Be, activation_function['np'], apply_activation_to_embedding=False, bn_variables=batchnorm_encode_variables)
else:
recon[partition], embed[partition], error[partition] = sdae_apply_functions.encode_and_decode(dataset[partition], W, Be, Bd, activation_function['np'], d['current_apply_activation_to_embedding'], d['apply_activation_to_output'], return_embedding=True, return_reconstruction_error=True)
embed_preactivation[partition] = sdae_apply_functions.encode(dataset[partition], W, Be, activation_function['np'], apply_activation_to_embedding=False)
print('{0} reconstruction error: {1:1.3g}'.format(partition, error[partition]), flush=True)
datasetIO.save_datamatrix('{0}/{1}_embedding_layer{2!s}_finetuning{3!s}.pickle'.format(d['output_path'], partition, d['current_hidden_layer'], d['current_finetuning_run']), embed[partition])
datasetIO.save_datamatrix('{0}/{1}_embedding_layer{2!s}_finetuning{3!s}.txt.gz'.format(d['output_path'], partition, d['current_hidden_layer'], d['current_finetuning_run']), embed[partition])
if d['current_apply_activation_to_embedding']:
datasetIO.save_datamatrix('{0}/{1}_embedding_preactivation_layer{2!s}_finetuning{3!s}.pickle'.format(d['output_path'], partition, d['current_hidden_layer'], d['current_finetuning_run']), embed_preactivation[partition])
datasetIO.save_datamatrix('{0}/{1}_embedding_preactivation_layer{2!s}_finetuning{3!s}.txt.gz'.format(d['output_path'], partition, d['current_hidden_layer'], d['current_finetuning_run']), embed_preactivation[partition])
# plot loss
print('plotting loss...', flush=True)
fg, ax = plt.subplots(1, 1, figsize=(3.25,2.25))
ax.set_position([0.55/3.25, 0.45/2.25, 2.6/3.25, 1.7/2.25])
ax.semilogx(reporting_steps, train_losses, ':r', linewidth=1, label='train')
ax.semilogx(reporting_steps, valid_losses, '-g', linewidth=1, label='valid')
ax.semilogx(reporting_steps, train_noisy_losses, '--b', linewidth=1, label='train,noisy')
ax.semilogx(reporting_steps, valid_noisy_losses, '-.k', linewidth=1, label='valid,noisy')
ax.legend(loc='best', fontsize=8)
ax.set_ylabel('loss', fontsize=8)
ax.set_xlabel('steps (selected step:{0!s})'.format(selected_step), fontsize=8)
ax.set_xlim(reporting_steps[0]-1, reporting_steps[-1]+1)
# ax.set_ylim(0, 1)
ax.tick_params(axis='both', which='major', left=True, right=True, bottom=True, top=False, labelleft=True, labelright=False, labelbottom=True, labeltop=False, labelsize=8)
fg.savefig('{0}/optimization_path_layer{1!s}_finetuning{2!s}.png'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run']), transparent=True, pad_inches=0, dpi=600)
plt.close()
# plot reconstructions
print('plotting reconstructions...', flush=True)
num_recons = min([d['reconstruction_rows']*d['reconstruction_cols'], dataset['valid'].shape[0]])
x_valid = dataset['valid'].matrix[:num_recons,:]
xr_valid = recon['valid'].matrix[:num_recons,:]
if x_valid.shape[1] > 1000:
x_valid = x_valid[:,:1000]
xr_valid = xr_valid[:,:1000]
lb = np.append(x_valid, xr_valid, 1).min(1)
ub = np.append(x_valid, xr_valid, 1).max(1)
fg, axs = plt.subplots(d['reconstruction_rows'], d['reconstruction_cols'], figsize=(6.5,3.25))
for i, ax in enumerate(axs.reshape(-1)):
if i < num_recons:
ax.plot(x_valid[i,:], xr_valid[i,:], 'ok', markersize=0.5, markeredgewidth=0)
ax.set_ylim(lb[i], ub[i])
ax.set_xlim(lb[i], ub[i])
ax.tick_params(axis='both', which='major', left=False, right=False, bottom=False, top=False, labelleft=False, labelright=False, labelbottom=False, labeltop=False, pad=4)
ax.set_frame_on(False)
ax.axvline(lb[i], linewidth=1, color='k')
ax.axvline(ub[i], linewidth=1, color='k')
ax.axhline(lb[i], linewidth=1, color='k')
ax.axhline(ub[i], linewidth=1, color='k')
else:
fg.delaxes(ax)
fg.savefig('{0}/reconstructions_layer{1!s}_finetuning{2!s}.png'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run']), transparent=True, pad_inches=0, dpi=1200)
plt.close()
# plot 2d embedding
if d['current_dimensions'][-1] == 2 and (not d['use_finetuning'] or d['current_finetuning_run'] > 0):
print('plotting 2d embedding...', flush=True)
fg, ax = plt.subplots(1, 1, figsize=(6.5,6.5))
ax.set_position([0.15/6.5, 0.15/6.5, 6.2/6.5, 6.2/6.5])
ax.plot(embed['train'].matrix[:,0], embed['train'].matrix[:,1], 'ok', markersize=2, markeredgewidth=0, alpha=0.5, zorder=0)
ax.plot(embed['valid'].matrix[:,0], embed['valid'].matrix[:,1], 'or', markersize=2, markeredgewidth=0, alpha=1.0, zorder=1)
ax.tick_params(axis='both', which='major', bottom=False, top=False, labelbottom=False, labeltop=False, left=False, right=False, labelleft=False, labelright=False, pad=4)
ax.set_frame_on(False)
fg.savefig('{0}/embedding_layer{1!s}_finetuning{2!s}.png'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run']), transparent=True, pad_inches=0, dpi=600)
plt.close()
if d['current_apply_activation_to_embedding']:
fg, ax = plt.subplots(1, 1, figsize=(6.5,6.5))
ax.set_position([0.15/6.5, 0.15/6.5, 6.2/6.5, 6.2/6.5])
ax.plot(embed_preactivation['train'].matrix[:,0], embed_preactivation['train'].matrix[:,1], 'ok', markersize=2, markeredgewidth=0, alpha=0.5, zorder=0)
ax.plot(embed_preactivation['valid'].matrix[:,0], embed_preactivation['valid'].matrix[:,1], 'or', markersize=2, markeredgewidth=0, alpha=1.0, zorder=1)
ax.tick_params(axis='both', which='major', bottom=False, top=False, labelbottom=False, labeltop=False, left=False, right=False, labelleft=False, labelright=False, pad=4)
ax.set_frame_on(False)
fg.savefig('{0}/embedding_preactivation_layer{1!s}_finetuning{2!s}.png'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run']), transparent=True, pad_inches=0, dpi=600)
plt.close()
# log selected step
with open('{0}/log.txt'.format(d['output_path']), mode='at', buffering=1) as fl:
fl.write('\nadjusted selected step:{0}\n'.format(selected_step))
print('done adjust_early_stopping.', flush=True)
def main(d):
# d is a dictionary containing the auto-encoder design specifications and training phase specifications
# RESET DEFAULT GRAPH
print('resetting default graph...', flush=True)
tf.reset_default_graph()
# FINISH CONFIGURATION
print('finishing configuration...', flush=True)
# specify noise distribution
if d['noise_distribution'] == 'truncnorm':
noise_distribution = tf.truncated_normal
elif d['noise_distribution'] == 'uniform':
noise_distribution = tf.random_uniform
# specify distribution of initial weights
if d['initialization_distribution'] == 'truncnorm':
initialization_distribution = tf.truncated_normal
# specify activation function
if d['activation_function'] == 'tanh':
activation_function = {'tf': tf.tanh, 'np': sdae_apply_functions.tanh}
elif d['activation_function'] == 'relu':
activation_function = {
'tf': tf.nn.relu,
'np': sdae_apply_functions.relu
}
elif d['activation_function'] == 'elu':
activation_function = {'tf': tf.nn.elu, 'np': sdae_apply_functions.elu}
elif d['activation_function'] == 'sigmoid':
activation_function = {
'tf': tf.sigmoid,
'np': sdae_apply_functions.sigmoid
}
# load data
partitions = ['train', 'valid', 'test']
dataset = {}
for partition in partitions:
dataset[partition] = datasetIO.load_datamatrix('{0}/{1}.pickle'.format(
d['input_path'], partition))
d['{0}_examples'.format(partition)] = dataset[partition].shape[0]
# create output directory
if not os.path.exists(d['output_path']):
os.makedirs(d['output_path'])
# initialize model architecture (number of layers and dimension of each layer)
d['current_dimensions'] = d[
'all_dimensions'][:d['current_hidden_layer'] +
1] # dimensions of model up to current depth
# specify embedding function for current training phase
# we want the option of skipping the embedding activation function to apply only to the full model
if not d['apply_activation_to_embedding'] and d['current_dimensions'] == d[
'all_dimensions']:
d['current_apply_activation_to_embedding'] = False
else:
d['current_apply_activation_to_embedding'] = True
# initialize assignments of training examples to mini-batches and number of training steps for stochastic gradient descent
d['batch_size'] = d['batch_fraction'] * d['train_examples']
batch_ids = create_batch_ids(d['train_examples'], d['batch_size'])
d['batches'] = np.unique(batch_ids).size
d['steps'] = d['current_epochs'] * d['batches']
# specify path to weights from previous training run
d['previous_variables_path'] = '{0}/variables_layer{1!s}_finetuning{2!s}.pickle'.format(
d['output_path'], d['previous_hidden_layer'],
d['previous_finetuning_run'])
d['fix_or_init'] = 'fix' if d[
'current_finetuning_run'] == 0 else 'init' # fix for pretraining, init for finetuning
# specify rows and columns of figure showing data reconstructions
d['reconstruction_rows'] = int(
np.round(np.sqrt(np.min([100, d['valid_examples']]) / 2)))
d['reconstruction_cols'] = 2 * d['reconstruction_rows']
# print some design information
print('input path: {0}'.format(d['input_path']), flush=True)
print('output path: {0}'.format(d['output_path']), flush=True)
print('previous variables path: {0}'.format(d['previous_variables_path']),
flush=True)
print('previous variables fix or init: {0}'.format(d['fix_or_init']),
flush=True)
# SAVE CURRENT DESIGN
print('saving current design...', flush=True)
with open('{0}/design_layer{1!s}_finetuning{2!s}.json'.format(
d['output_path'], d['current_hidden_layer'],
d['current_finetuning_run']),
mode='wt',
encoding='utf-8',
errors='surrogateescape') as fw:
json.dump(d, fw, indent=2)
# DEFINE REPORTING VARIABLES
print('defining reporting variables...', flush=True)
reporting_steps = sdae_design_functions.create_reporting_steps(
d['steps'], d['firstcheckpoint'], d['maxstepspercheckpoint'])
valid_losses = np.zeros(reporting_steps.size, dtype='float32')
train_losses = np.zeros(reporting_steps.size, dtype='float32')
valid_noisy_losses = np.zeros(reporting_steps.size, dtype='float32')
train_noisy_losses = np.zeros(reporting_steps.size, dtype='float32')
print('reporting steps:', reporting_steps, flush=True)
# DEFINE COMPUTATIONAL GRAPH
# define placeholders for input data, use None to allow feeding different numbers of examples
print('defining placeholders...', flush=True)
noise_stdv = tf.placeholder(tf.float32, [])
noise_prob = tf.placeholder(tf.float32, [])
training_and_validation_data_initializer = tf.placeholder(
tf.float32, [
dataset['train'].shape[0] + dataset['valid'].shape[0],
dataset['train'].shape[1]
])
selection_mask = tf.placeholder(
tf.bool, [dataset['train'].shape[0] + dataset['valid'].shape[0]])
# define variables
# W contains the weights, bencode contains the biases for encoding, and bdecode contains the biases for decoding
print('defining variables...', flush=True)
training_and_validation_data = tf.Variable(
training_and_validation_data_initializer,
trainable=False,
collections=[])
if os.path.exists(d['previous_variables_path']):
# update variables (if continuing from a previous training run)
print('loading previous variables...', flush=True)
global_step, W, bencode, bdecode = update_variables(
d['current_dimensions'], initialization_distribution,
d['initialization_sigma'], d['previous_variables_path'],
d['fix_or_init'], d['include_global_step'])
elif d['current_hidden_layer'] == 1 and d['current_finetuning_run'] == 0:
# create variables
global_step, W, bencode, bdecode = create_variables(
d['current_dimensions'], initialization_distribution,
d['initialization_sigma'])
else:
raise ValueError('could not find previous variables')
# define model
# h contains the activations from input layer to bottleneck layer
# hhat contains the activations from bottleneck layer to output layer
# xhat is a reference to the output layer (i.e. the reconstruction)
print('defining model...', flush=True)
x = tf.boolean_mask(training_and_validation_data, selection_mask)
if d['noise_distribution'] == 'truncnorm':
noise = noise_distribution(tf.shape(x), stddev=noise_stdv)
else:
noise = noise_distribution(tf.shape(x),
minval=-noise_stdv,
maxval=noise_stdv)
noise_mask = tf.to_float(tf.random_uniform(tf.shape(x)) <= noise_prob)
xnoisy = apply_noise(x, noise, noise_mask, d['noise_operation'])
h, hhat, xhat = create_autoencoder(
xnoisy, activation_function['tf'], d['apply_activation_to_output'],
d['current_apply_activation_to_embedding'], W, bencode, bdecode)
# define loss
print('defining loss...', flush=True)
loss = tf.reduce_mean(tf.squared_difference(x, xhat)) # squared error loss
# define optimizer and training function
print('defining optimizer and training function...', flush=True)
optimizer = tf.train.AdamOptimizer(learning_rate=d['learning_rate'],
epsilon=d['epsilon'],
beta1=d['beta1'],
beta2=d['beta2'])
train_fn = optimizer.minimize(loss, global_step=global_step)
# define bottleneck layer preactivation
# bottleneck_preactivation = tf.matmul(h[-2], W[-1]) + bencode[-1]
# INITIALIZE TENSORFLOW SESSION
print('initializing tensorflow session...', flush=True)
init = tf.global_variables_initializer()
session_config = configure_session(d['processor'],
d['gpu_memory_fraction'])
with tf.Session(config=session_config) as sess:
sess.run(init)
# TRAINING
print('training...', flush=True)
sess.run(training_and_validation_data.initializer,
feed_dict={
training_and_validation_data_initializer:
np.append(dataset['train'].matrix,
dataset['valid'].matrix, 0)
})
validation_id = -1
batch_and_validation_ids = np.full(dataset['train'].shape[0] +
dataset['valid'].shape[0],
validation_id,
dtype=batch_ids.dtype)
is_train = np.append(np.ones(dataset['train'].shape[0], dtype='bool'),
np.zeros(dataset['valid'].shape[0], dtype='bool'))
is_valid = ~is_train
training_step = 0
i = 0
overfitting_score = 0
stopearly = False
starttime = time.time()
with open('{0}/log_layer{1!s}_finetuning{2!s}.txt'.format(
d['output_path'], d['current_hidden_layer'],
d['current_finetuning_run']),
mode='wt',
buffering=1) as fl:
fl.write('\t'.join([
'step', 'train_loss', 'valid_loss', 'train_noisy_loss',
'valid_noisy_loss', 'time'
]) + '\n')
for epoch in range(d['current_epochs']):
if stopearly:
break
# randomize assignment of training examples to batches
np.random.shuffle(batch_ids)
batch_and_validation_ids[is_train] = batch_ids
for batch in range(d['batches']):
training_step += 1
# select mini-batch
selected = batch_and_validation_ids == batch
# update weights
sess.run(train_fn,
feed_dict={
selection_mask: selected,
noise_prob: d['noise_probability'],
noise_stdv: d['noise_sigma']
})
# record training and validation errors
if training_step == reporting_steps[i]:
train_losses[i] = sess.run(loss,
feed_dict={
selection_mask:
is_train,
noise_prob: 0,
noise_stdv: 0
})
train_noisy_losses[i] = sess.run(
loss,
feed_dict={
selection_mask: is_train,
noise_prob: d['noise_probability'],
noise_stdv: d['noise_sigma']
})
valid_losses[i] = sess.run(loss,
feed_dict={
selection_mask:
is_valid,
noise_prob: 0,
noise_stdv: 0
})
valid_noisy_losses[i] = sess.run(
loss,
feed_dict={
selection_mask: is_valid,
noise_prob: d['noise_probability'],
noise_stdv: d['noise_sigma']
})
print(
'step:{0:1.6g}, train loss:{1:1.3g}, valid loss:{2:1.3g}, train noisy loss:{3:1.3g},valid noisy loss:{4:1.3g}, time:{5:1.6g}'
.format(reporting_steps[i], train_losses[i],
valid_losses[i], train_noisy_losses[i],
valid_noisy_losses[i],
time.time() - starttime),
flush=True)
fl.write('\t'.join([
'{0:1.6g}'.format(x) for x in [
reporting_steps[i], train_losses[i],
valid_losses[i], train_noisy_losses[i],
valid_noisy_losses[i],
time.time() - starttime
]
]) + '\n')
# save current weights, reconstructions, and projections
if training_step >= d[
'startsavingstep'] or training_step == reporting_steps[
-1]:
with open(
'{0}/intermediate_variables_layer{1!s}_finetuning{2!s}_step{3!s}.pickle'
.format(d['output_path'],
d['current_hidden_layer'],
d['current_finetuning_run'],
training_step), 'wb') as fw:
pickle.dump(
(sess.run(global_step), sess.run(W),
sess.run(bencode), sess.run(bdecode)), fw)
# stop early if overfitting
if valid_losses[i] >= 1.01 * (np.insert(
valid_losses[:i], 0, np.inf).min()):
overfitting_score += 1
else:
overfitting_score = 0
if overfitting_score == d['overfitting_score_max']:
stopearly = True
print('stopping early!', flush=True)
break
i += 1
# end tensorflow session
print('closing tensorflow session...', flush=True)
# ROLL BACK IF OVERFITTING
if stopearly:
print('rolling back...', flush=True)
reporting_steps = reporting_steps[:i + 1]
train_losses = train_losses[:i + 1]
valid_losses = valid_losses[:i + 1]
train_noisy_losses = train_noisy_losses[:i + 1]
valid_noisy_losses = valid_noisy_losses[:i + 1]
# selected_step = max([reporting_steps[i-d['overfitting_score_max']], d['startsavingstep']])
else:
print('completed all training steps...', flush=True)
# selected_step = reporting_steps[-1]
selected_step = min([
max([reporting_steps[np.argmin(valid_losses)], d['startsavingstep']]),
reporting_steps[-1]
])
print('selected step:{0}...'.format(selected_step), flush=True)
# SAVE RESULTS
print('saving results...', flush=True)
with open(
'{0}/optimization_path_layer{1!s}_finetuning{2!s}.pickle'.format(
d['output_path'], d['current_hidden_layer'],
d['current_finetuning_run']), 'wb') as fw:
pickle.dump(
{
'reporting_steps': reporting_steps,
'valid_losses': valid_losses,
'train_losses': train_losses,
'valid_noisy_losses': valid_noisy_losses,
'train_noisy_losses': train_noisy_losses
}, fw)
if d['current_dimensions'] == d['all_dimensions'] and (
not d['use_finetuning'] or d['current_finetuning_run'] > 0):
shutil.copyfile(
'{0}/intermediate_variables_layer{1!s}_finetuning{2!s}_step{3!s}.pickle'
.format(d['output_path'], d['current_hidden_layer'],
d['current_finetuning_run'], selected_step),
'{0}/variables_layer{1!s}_finetuning{2!s}.pickle'.format(
d['output_path'], d['current_hidden_layer'],
d['current_finetuning_run']))
else:
shutil.move(
'{0}/intermediate_variables_layer{1!s}_finetuning{2!s}_step{3!s}.pickle'
.format(d['output_path'], d['current_hidden_layer'],
d['current_finetuning_run'], selected_step),
'{0}/variables_layer{1!s}_finetuning{2!s}.pickle'.format(
d['output_path'], d['current_hidden_layer'],
d['current_finetuning_run']))
with open(
'{0}/variables_layer{1!s}_finetuning{2!s}.pickle'.format(
d['output_path'], d['current_hidden_layer'],
d['current_finetuning_run']), 'rb') as fr:
W, Be, Bd = pickle.load(fr)[1:] # global_step, W, bencode, bdecode
recon = {}
embed = {}
error = {}
embed_preactivation = {}
for partition in partitions:
recon[partition], embed[partition], error[
partition] = sdae_apply_functions.encode_and_decode(
dataset[partition],
W,
Be,
Bd,
activation_function['np'],
d['current_apply_activation_to_embedding'],
d['apply_activation_to_output'],
return_embedding=True,
return_reconstruction_error=True)
embed_preactivation[partition] = sdae_apply_functions.encode(
dataset[partition],
W,
Be,
activation_function['np'],
apply_activation_to_embedding=False)
print('{0} reconstruction error: {1:1.3g}'.format(
partition, error[partition]),
flush=True)
if d['current_dimensions'] == d['all_dimensions'] and (
not d['use_finetuning'] or d['current_finetuning_run'] > 0):
datasetIO.save_datamatrix(
'{0}/{1}_embedding_layer{2!s}_finetuning{3!s}.pickle'.format(
d['output_path'], partition, d['current_hidden_layer'],
d['current_finetuning_run']), embed[partition])
datasetIO.save_datamatrix(
'{0}/{1}_embedding_layer{2!s}_finetuning{3!s}.txt.gz'.format(
d['output_path'], partition, d['current_hidden_layer'],
d['current_finetuning_run']), embed[partition])
if d['current_apply_activation_to_embedding']:
datasetIO.save_datamatrix(
'{0}/{1}_embedding_preactivation_layer{2!s}_finetuning{3!s}.pickle'
.format(d['output_path'], partition,
d['current_hidden_layer'],
d['current_finetuning_run']),
embed_preactivation[partition])
datasetIO.save_datamatrix(
'{0}/{1}_embedding_preactivation_layer{2!s}_finetuning{3!s}.txt.gz'
.format(d['output_path'], partition,
d['current_hidden_layer'],
d['current_finetuning_run']),
embed_preactivation[partition])
# PLOT LOSS
print('plotting loss...', flush=True)
fg, ax = plt.subplots(1, 1, figsize=(3.25, 2.25))
ax.set_position([0.55 / 3.25, 0.45 / 2.25, 2.6 / 3.25, 1.7 / 2.25])
ax.semilogx(reporting_steps,
train_losses,
':r',
linewidth=1,
label='train')
ax.semilogx(reporting_steps,
valid_losses,
'-g',
linewidth=1,
label='valid')
ax.semilogx(reporting_steps,
train_noisy_losses,
'--b',
linewidth=1,
label='train,noisy')
ax.semilogx(reporting_steps,
valid_noisy_losses,
'-.k',
linewidth=1,
label='valid,noisy')
ax.legend(loc='best', fontsize=8)
ax.set_ylabel('loss', fontsize=8)
ax.set_xlabel('steps (selected step:{0!s})'.format(selected_step),
fontsize=8)
ax.set_xlim(reporting_steps[0] - 1, reporting_steps[-1] + 1)
# ax.set_ylim(0, 1)
ax.tick_params(axis='both',
which='major',
left='on',
right='on',
bottom='on',
top='off',
labelleft='on',
labelright='off',
labelbottom='on',
labeltop='off',
labelsize=8)
fg.savefig('{0}/optimization_path_layer{1!s}_finetuning{2!s}.png'.format(
d['output_path'], d['current_hidden_layer'],
d['current_finetuning_run']),
transparent=True,
pad_inches=0,
dpi=600)
plt.close()
# PLOT RECONSTRUCTIONS
print('plotting reconstructions...', flush=True)
x_valid = dataset['valid'].matrix[:d['reconstruction_rows'] *
d['reconstruction_cols'], :]
xr_valid = recon['valid'].matrix[:d['reconstruction_rows'] *
d['reconstruction_cols'], :]
if x_valid.shape[1] > 1000:
x_valid = x_valid[:, :1000]
xr_valid = xr_valid[:, :1000]
lb = np.append(x_valid, xr_valid, 1).min(1)
ub = np.append(x_valid, xr_valid, 1).max(1)
fg, axs = plt.subplots(d['reconstruction_rows'],
d['reconstruction_cols'],
figsize=(6.5, 3.25))
for i, ax in enumerate(axs.reshape(-1)):
ax.plot(x_valid[i, :],
xr_valid[i, :],
'ok',
markersize=0.5,
markeredgewidth=0)
ax.set_ylim(lb[i], ub[i])
ax.set_xlim(lb[i], ub[i])
ax.tick_params(axis='both',
which='major',
left='off',
right='off',
bottom='off',
top='off',
labelleft='off',
labelright='off',
labelbottom='off',
labeltop='off',
pad=4)
ax.set_frame_on(False)
ax.axvline(lb[i], linewidth=1, color='k')
ax.axvline(ub[i], linewidth=1, color='k')
ax.axhline(lb[i], linewidth=1, color='k')
ax.axhline(ub[i], linewidth=1, color='k')
fg.savefig('{0}/reconstructions_layer{1!s}_finetuning{2!s}.png'.format(
d['output_path'], d['current_hidden_layer'],
d['current_finetuning_run']),
transparent=True,
pad_inches=0,
dpi=1200)
plt.close()
# PLOT 2D EMBEDDING
if d['current_dimensions'][-1] == 2 and (not d['use_finetuning'] or
d['current_finetuning_run'] > 0):
print('plotting 2d embedding...', flush=True)
fg, ax = plt.subplots(1, 1, figsize=(6.5, 6.5))
ax.set_position([0.15 / 6.5, 0.15 / 6.5, 6.2 / 6.5, 6.2 / 6.5])
ax.plot(embed['train'].matrix[:, 0],
embed['train'].matrix[:, 1],
'ok',
markersize=2,
markeredgewidth=0,
alpha=0.5,
zorder=0)
ax.plot(embed['valid'].matrix[:, 0],
embed['valid'].matrix[:, 1],
'or',
markersize=2,
markeredgewidth=0,
alpha=1.0,
zorder=1)
ax.tick_params(axis='both',
which='major',
bottom='off',
top='off',
labelbottom='off',
labeltop='off',
left='off',
right='off',
labelleft='off',
labelright='off',
pad=4)
ax.set_frame_on(False)
fg.savefig('{0}/embedding_layer{1!s}_finetuning{2!s}.png'.format(
d['output_path'], d['current_hidden_layer'],
d['current_finetuning_run']),
transparent=True,
pad_inches=0,
dpi=600)
plt.close()
if d['current_apply_activation_to_embedding']:
fg, ax = plt.subplots(1, 1, figsize=(6.5, 6.5))
ax.set_position([0.15 / 6.5, 0.15 / 6.5, 6.2 / 6.5, 6.2 / 6.5])
ax.plot(embed_preactivation['train'].matrix[:, 0],
embed_preactivation['train'].matrix[:, 1],
'ok',
markersize=2,
markeredgewidth=0,
alpha=0.5,
zorder=0)
ax.plot(embed_preactivation['valid'].matrix[:, 0],
embed_preactivation['valid'].matrix[:, 1],
'or',
markersize=2,
markeredgewidth=0,
alpha=1.0,
zorder=1)
ax.tick_params(axis='both',
which='major',
bottom='off',
top='off',
labelbottom='off',
labeltop='off',
left='off',
right='off',
labelleft='off',
labelright='off',
pad=4)
ax.set_frame_on(False)
fg.savefig(
'{0}/embedding_preactivation_layer{1!s}_finetuning{2!s}.png'.
format(d['output_path'], d['current_hidden_layer'],
d['current_finetuning_run']),
transparent=True,
pad_inches=0,
dpi=600)
plt.close()
print('done training phase.', flush=True)
return d['current_hidden_layer'], d['current_finetuning_run'], d[
'current_epochs']
def main(visualizations_path):
# read visualizations
print('reading visualizations...', flush=True)
designpath_selectedstep = []
with open(visualizations_path, mode='rt', encoding='utf-8', errors='surrogateescape') as fr:
for line in fr:
designpath_selectedstep.append([x.strip() for x in line.split('\t')])
print('found {0!s} visualizations...'.format(len(designpath_selectedstep)), flush=True)
# make visualizations
print('making visualizations...', flush=True)
for didx, (design_path, selected_step) in enumerate(designpath_selectedstep):
print('working on {0}...'.format(design_path), flush=True)
print('selected step:{0!s}...'.format(selected_step), flush=True)
# load design
print('loading design...', flush=True)
with open(design_path, mode='rt', encoding='utf-8', errors='surrogateescape') as fr:
d = json.load(fr)
if 'apply_activation_to_embedding' not in d: # for legacy code
d['apply_activation_to_embedding'] = True
if 'use_batchnorm' not in d: # for legacy code
d['use_batchnorm'] = False
if 'skip_layerwise_training' not in d: # for legacy code
d['skip_layerwise_training'] = False
phase = d['training_schedule'][-1]
d['current_hidden_layer'] = phase['hidden_layer']
d['current_finetuning_run'] = phase['finetuning_run']
d['current_epochs'] = phase['epochs']
# load data
if didx == 0:
print('loading data...', flush=True)
partitions = ['train', 'valid', 'test']
dataset = {}
for partition in partitions:
dataset[partition] = datasetIO.load_datamatrix('{0}/{1}.pickle'.format(d['input_path'], partition))
# finish configuration
print('finishing configuration...', flush=True)
# specify activation function
if d['activation_function'] == 'tanh':
activation_function = {'np':sdae_apply_functions.tanh}
elif d['activation_function'] == 'relu':
activation_function = {'np':sdae_apply_functions.relu}
elif d['activation_function'] == 'elu':
activation_function = {'np':sdae_apply_functions.elu}
elif d['activation_function'] == 'sigmoid':
activation_function = {'np':sdae_apply_functions.sigmoid}
# initialize model architecture (number of layers and dimension of each layer)
d['current_dimensions'] = d['all_dimensions'][:d['current_hidden_layer']+1] # dimensions of model up to current depth
# specify embedding function for current training phase
# we want the option of skipping the embedding activation function to apply only to the full model
if not d['apply_activation_to_embedding'] and d['current_dimensions'] == d['all_dimensions']:
d['current_apply_activation_to_embedding'] = False
else:
d['current_apply_activation_to_embedding'] = True
print('current_apply_activation_to_embedding: {0!s}'.format(d['current_apply_activation_to_embedding']), flush=True)
# specify rows and columns of figure showing data reconstructions
d['reconstruction_rows'] = int(np.round(np.sqrt(np.min([100, dataset['valid'].shape[0]])/2)))
d['reconstruction_cols'] = 2*d['reconstruction_rows']
# load model variables
print('loading model variables...', flush=True)
with open('{0}/intermediate_variables_layer{1!s}_finetuning{2!s}_step{3!s}.pickle'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run'], selected_step), 'rb') as fr:
W, Be, Bd = pickle.load(fr)[1:] # global_step, W, bencode, bdecode
if d['use_batchnorm']:
with open('{0}/intermediate_batchnorm_variables_layer{1!s}_finetuning{2!s}_step{3!s}.pickle'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run'], selected_step), 'rb') as fr:
batchnorm_variables = pickle.load(fr) # gammas, betas, moving_means, moving_variances
batchnorm_encode_variables, batchnorm_decode_variables = sdae_apply_functions.align_batchnorm_variables(batchnorm_variables, d['current_apply_activation_to_embedding'], d['apply_activation_to_output'])
# compute embedding and reconstruction
print('computing embedding and reconstruction...', flush=True)
recon = {}
embed = {}
error = {}
embed_preactivation = {}
for partition in partitions:
if d['use_batchnorm']:
recon[partition], embed[partition], error[partition] = sdae_apply_functions.encode_and_decode(dataset[partition], W, Be, Bd, activation_function['np'], d['current_apply_activation_to_embedding'], d['apply_activation_to_output'], return_embedding=True, return_reconstruction_error=True, bn_encode_variables=batchnorm_encode_variables, bn_decode_variables=batchnorm_decode_variables)
embed_preactivation[partition] = sdae_apply_functions.encode(dataset[partition], W, Be, activation_function['np'], apply_activation_to_embedding=False, bn_variables=batchnorm_encode_variables)
else:
recon[partition], embed[partition], error[partition] = sdae_apply_functions.encode_and_decode(dataset[partition], W, Be, Bd, activation_function['np'], d['current_apply_activation_to_embedding'], d['apply_activation_to_output'], return_embedding=True, return_reconstruction_error=True)
embed_preactivation[partition] = sdae_apply_functions.encode(dataset[partition], W, Be, activation_function['np'], apply_activation_to_embedding=False)
if 'all' not in embed:
embed['all'] = copy.deepcopy(embed[partition])
recon['all'] = copy.deepcopy(recon[partition])
embed_preactivation['all'] = copy.deepcopy(embed_preactivation[partition])
else:
embed['all'].append(embed[partition], 0)
recon['all'].append(recon[partition], 0)
embed_preactivation['all'].append(embed_preactivation[partition], 0)
print('{0} reconstruction error: {1:1.3g}'.format(partition, error[partition]), flush=True)
datasetIO.save_datamatrix('{0}/{1}_intermediate_embedding_layer{2!s}_finetuning{3!s}_step{4!s}.pickle'.format(d['output_path'], partition, d['current_hidden_layer'], d['current_finetuning_run'], selected_step), embed[partition])
datasetIO.save_datamatrix('{0}/{1}_intermediate_embedding_layer{2!s}_finetuning{3!s}_step{4!s}.txt.gz'.format(d['output_path'], partition, d['current_hidden_layer'], d['current_finetuning_run'], selected_step), embed[partition])
if d['current_apply_activation_to_embedding']:
datasetIO.save_datamatrix('{0}/{1}_intermediate_embedding_preactivation_layer{2!s}_finetuning{3!s}_step{4!s}.pickle'.format(d['output_path'], partition, d['current_hidden_layer'], d['current_finetuning_run'], selected_step), embed_preactivation[partition])
datasetIO.save_datamatrix('{0}/{1}_intermediate_embedding_preactivation_layer{2!s}_finetuning{3!s}_step{4!s}.txt.gz'.format(d['output_path'], partition, d['current_hidden_layer'], d['current_finetuning_run'], selected_step), embed_preactivation[partition])
embed['valid_test'] = copy.deepcopy(embed['valid'])
embed['valid_test'].append(embed['test'], 0)
embed_preactivation['valid_test'] = copy.deepcopy(embed_preactivation['valid'])
embed_preactivation['valid_test'].append(embed_preactivation['test'], 0)
# plot reconstructions
print('plotting reconstructions...', flush=True)
num_recons = min([d['reconstruction_rows']*d['reconstruction_cols'], dataset['valid'].shape[0]])
x_valid = dataset['valid'].matrix[:num_recons,:]
xr_valid = recon['valid'].matrix[:num_recons,:]
if x_valid.shape[1] > 1000:
x_valid = x_valid[:,:1000]
xr_valid = xr_valid[:,:1000]
lb = np.append(x_valid, xr_valid, 1).min(1)
ub = np.append(x_valid, xr_valid, 1).max(1)
fg, axs = plt.subplots(d['reconstruction_rows'], d['reconstruction_cols'], figsize=(6.5,3.25))
for i, ax in enumerate(axs.reshape(-1)):
if i < num_recons:
ax.plot(x_valid[i,:], xr_valid[i,:], 'ok', markersize=0.5, markeredgewidth=0)
ax.set_ylim(lb[i], ub[i])
ax.set_xlim(lb[i], ub[i])
ax.tick_params(axis='both', which='major', left=False, right=False, bottom=False, top=False, labelleft=False, labelright=False, labelbottom=False, labeltop=False, pad=4)
ax.set_frame_on(False)
ax.axvline(lb[i], linewidth=1, color='k')
ax.axvline(ub[i], linewidth=1, color='k')
ax.axhline(lb[i], linewidth=1, color='k')
ax.axhline(ub[i], linewidth=1, color='k')
else:
fg.delaxes(ax)
fg.savefig('{0}/intermediate_reconstructions_layer{1!s}_finetuning{2!s}_step{3!s}.png'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run'], selected_step), transparent=True, pad_inches=0, dpi=1200)
plt.close()
# plot 2d embedding
if d['current_dimensions'][-1] == 2:
print('plotting 2d embedding...', flush=True)
fg, ax = plt.subplots(1, 1, figsize=(6.5,6.5))
ax.set_position([0.15/6.5, 0.15/6.5, 6.2/6.5, 6.2/6.5])
ax.plot(embed['train'].matrix[:,0], embed['train'].matrix[:,1], 'ok', markersize=2, markeredgewidth=0, alpha=0.5, zorder=0)
ax.plot(embed['valid'].matrix[:,0], embed['valid'].matrix[:,1], 'or', markersize=2, markeredgewidth=0, alpha=1.0, zorder=1)
ax.plot(embed['test'].matrix[:,0], embed['test'].matrix[:,1], 'ob', markersize=2, markeredgewidth=0, alpha=1.0, zorder=2)
ax.tick_params(axis='both', which='major', bottom=False, top=False, labelbottom=False, labeltop=False, left=False, right=False, labelleft=False, labelright=False, pad=4)
ax.set_frame_on(False)
fg.savefig('{0}/intermediate_embedding_layer{1!s}_finetuning{2!s}_step{3!s}.png'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run'], selected_step), transparent=True, pad_inches=0, dpi=600)
plt.close()
fields = ['general_tissue']
parts = ['all', 'train', 'valid', 'test', 'valid_test']
for field in fields:
for partition in parts:
values = np.unique(embed[partition].rowmeta[field])
values = values[values != '-666']
abbrevs = np.array([x[:2] for x in values], dtype='object')
cmap = plt.get_cmap('gist_rainbow')
colors = [cmap(float((i+0.5)/len(values))) for i in range(len(values))]
zorder = 1
alpha = 0.5
fg, ax = plt.subplots(1, 1, figsize=(6.5,4.3))
ax.set_position([0.15/6.5, 0.15/4.3, 4.0/6.5, 4.0/4.3])
for value, abbrev, color in zip(values, abbrevs, colors):
hit = embed[partition].rowmeta[field] == value
hidxs = hit.nonzero()[0]
ax.plot(embed[partition].matrix[hit,0], embed[partition].matrix[hit,1], linestyle='None', linewidth=0, marker='o', markerfacecolor=color, markeredgecolor=color, markersize=0.2, markeredgewidth=0, alpha=alpha, zorder=zorder, label='{0}, {1}'.format(abbrev, value))
for hidx in hidxs:
ax.text(embed[partition].matrix[hidx,0], embed[partition].matrix[hidx,1], abbrev, horizontalalignment='center', verticalalignment='center', fontsize=4, color=color, alpha=alpha, zorder=zorder, label='{0}, {1}'.format(abbrev, value))
ax.set_xlim(embed[partition].matrix[:,0].min(), embed[partition].matrix[:,0].max())
ax.set_ylim(embed[partition].matrix[:,1].min(), embed[partition].matrix[:,1].max())
ax.legend(loc='center left', bbox_to_anchor=(1, 0.5), borderaxespad=0, frameon=False, ncol=1, numpoints=1, markerscale=40, fontsize=8, labelspacing=0.25)
ax.tick_params(axis='both', which='major', bottom=False, top=False, labelbottom=False, labeltop=False, left=False, right=False, labelleft=False, labelright=False, pad=4)
ax.set_frame_on(False)
if partition == 'all':
partsuff = ''
else:
partsuff = '_' + partition
fg.savefig('{0}/intermediate_embedding_layer{1!s}_finetuning{2!s}_step{3!s}_coloredby_{4}{5}.png'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run'], selected_step, field, partsuff), transparent=True, pad_inches=0, dpi=600)
plt.close()
if d['current_apply_activation_to_embedding']:
fg, ax = plt.subplots(1, 1, figsize=(6.5,6.5))
ax.set_position([0.15/6.5, 0.15/6.5, 6.2/6.5, 6.2/6.5])
ax.plot(embed_preactivation['train'].matrix[:,0], embed_preactivation['train'].matrix[:,1], 'ok', markersize=2, markeredgewidth=0, alpha=0.5, zorder=0)
ax.plot(embed_preactivation['valid'].matrix[:,0], embed_preactivation['valid'].matrix[:,1], 'or', markersize=2, markeredgewidth=0, alpha=1.0, zorder=1)
ax.plot(embed_preactivation['test'].matrix[:,0], embed_preactivation['test'].matrix[:,1], 'ob', markersize=2, markeredgewidth=0, alpha=1.0, zorder=2)
ax.tick_params(axis='both', which='major', bottom=False, top=False, labelbottom=False, labeltop=False, left=False, right=False, labelleft=False, labelright=False, pad=4)
ax.set_frame_on(False)
fg.savefig('{0}/intermediate_embedding_preactivation_layer{1!s}_finetuning{2!s}_step{3!s}.png'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run'], selected_step), transparent=True, pad_inches=0, dpi=600)
plt.close()
fields = ['general_tissue']
parts = ['all', 'train', 'valid', 'test', 'valid_test']
for field in fields:
for partition in parts:
values = np.unique(embed_preactivation[partition].rowmeta[field])
values = values[values != '-666']
abbrevs = np.array([x[:2] for x in values], dtype='object')
cmap = plt.get_cmap('gist_rainbow')
colors = [cmap(float((i+0.5)/len(values))) for i in range(len(values))]
zorder = 1
alpha = 0.5
fg, ax = plt.subplots(1, 1, figsize=(6.5,4.3))
ax.set_position([0.15/6.5, 0.15/4.3, 4.0/6.5, 4.0/4.3])
for value, abbrev, color in zip(values, abbrevs, colors):
hit = embed_preactivation[partition].rowmeta[field] == value
hidxs = hit.nonzero()[0]
ax.plot(embed_preactivation[partition].matrix[hit,0], embed_preactivation[partition].matrix[hit,1], linestyle='None', linewidth=0, marker='o', markerfacecolor=color, markeredgecolor=color, markersize=0.2, markeredgewidth=0, alpha=alpha, zorder=zorder, label='{0}, {1}'.format(abbrev, value))
for hidx in hidxs:
ax.text(embed_preactivation[partition].matrix[hidx,0], embed_preactivation[partition].matrix[hidx,1], abbrev, horizontalalignment='center', verticalalignment='center', fontsize=4, color=color, alpha=alpha, zorder=zorder, label='{0}, {1}'.format(abbrev, value))
ax.set_xlim(embed_preactivation[partition].matrix[:,0].min(), embed_preactivation[partition].matrix[:,0].max())
ax.set_ylim(embed_preactivation[partition].matrix[:,1].min(), embed_preactivation[partition].matrix[:,1].max())
ax.legend(loc='center left', bbox_to_anchor=(1, 0.5), borderaxespad=0, frameon=False, ncol=1, numpoints=1, markerscale=40, fontsize=8, labelspacing=0.25)
ax.tick_params(axis='both', which='major', bottom=False, top=False, labelbottom=False, labeltop=False, left=False, right=False, labelleft=False, labelright=False, pad=4)
ax.set_frame_on(False)
if partition == 'all':
partsuff = ''
else:
partsuff = '_' + partition
fg.savefig('{0}/intermediate_embedding_preactivation_layer{1!s}_finetuning{2!s}_step{3!s}_coloredby_{4}{5}.png'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run'], selected_step, field, partsuff), transparent=True, pad_inches=0, dpi=600)
plt.close()
# plot heatmap
else:
print('plotting embedding heatmap...', flush=True)
for partition in partitions:
if 'all' not in embed:
embed['all'] = copy.deepcopy(embed[partition])
else:
embed['all'].append(embed[partition], 0)
embed['all'].cluster('all', 'cosine', 'average')
embed['all'].heatmap(rowmetalabels=[], columnmetalabels=[], normalize=False, standardize=False, normalizebeforestandardize=True, cmap_name='bwr', ub=None, lb=None, savefilename='{0}/intermediate_embedding_heatmap_layer{1!s}_finetuning{2!s}_step{3!s}.png'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run'], selected_step), closefigure=True, dpi=300)
if d['current_apply_activation_to_embedding']:
for partition in partitions:
if 'all' not in embed_preactivation:
embed_preactivation['all'] = copy.deepcopy(embed_preactivation[partition])
else:
embed_preactivation['all'].append(embed_preactivation[partition], 0)
embed_preactivation['all'].cluster('all', 'cosine', 'average')
embed_preactivation['all'].heatmap(rowmetalabels=[], columnmetalabels=[], normalize=False, standardize=False, normalizebeforestandardize=True, cmap_name='bwr', ub=None, lb=None, savefilename='{0}/intermediate_embedding_preactivation_heatmap_layer{1!s}_finetuning{2!s}_step{3!s}.png'.format(d['output_path'], d['current_hidden_layer'], d['current_finetuning_run'], selected_step), closefigure=True, dpi=300)
print('done get_sdae_features.', flush=True)