def plot_w1(self, annotate=False, highlight_syms=[], iter_num=None):
plt.clf()
configs = self.configs
if iter_num is not None:
fname_tag = 'iter_%d' % iter_num
else:
fname_tag = ''
from plot_utilities import plot_vec
# print self.W1.value.shape, len(self.data.syms)
assert self.W1.value.shape[0] == len(self.data.syms)
plot_vec(self.W1.value, self.data.syms, configs,
highlight_syms=highlight_syms,
doPCA=configs['do_pca'],
with_annotations=annotate,
fname_tag=fname_tag,
save=True)
def get_proxy_model():
fname = None
print fname
assert fname is not None, "Error: no model to retrieve in the time being"
with open(fname, "rb") as p:
w1 = pickle.load(p)
syms = pickle.load(p)
model_loss = pickle.load(p)
model_weights = pickle.load(p)
# configs_reloaded = pickle.load(p)
COMMON_TONE_CIRCLE = ["I", "iii", "V", "viio6", "ii", "IV", "vi", "I"]
circle_name = COMMON_TONE_CIRCLE
song_dict = make_song_dict(circle_name)
configs_dummy = get_configs()
fname, ax, vecs = plot_vec(w1, syms, configs_dummy, save=True, doPCA=True, return_ax=True)
add_arrow_annotation(syms, vecs, song_dict, ax, False)
plt.savefig("w1-%s.pdf" % Configs.get_timestamp())
model = WrappeSkipGram(w1, syms)
return model
