'help': 'Regularization for the matrix sqrt algorithm'
}],
[('--reg2', ), {
'type': float,
'default': 1E-8,
'help': 'Regularization for the gradient of the bures metric'
}],
[('--log_period', '-lp'), {
'type': int,
'default': 50,
'help': 'Logging period'
}]]
argparser = Parser("Deep Elliptical Embeddings")
argparser.add_arguments(args)
opt = argparser.get_dictionary()
viz = Visdom(port=8098)
vm = VisualManager(viz, 'marco')
root = r'/mnt/DATA/Prob_IR/'
context_dataset_name = r'context_data'
encoded_docs_filename = r'encoded_docs_model'
word_index_filename = r'word_index'
emb_filename = r'embeddings_dim_' + str(opt['dim']) + '_margin_' + str(
opt['margin'])
emb_path = os.path.join(root, emb_filename)
context_dataset_path = os.path.join(root, context_dataset_name)
print("Loading data...")
