alphas

alphas_again.py

def plot_key_alphas(data, softmax=True, fixed_axis=True):
  f, axes = pyplot.subplots(1, 3)
  f.set_size_inches((15, 2))
  for k in range(3):
    arr = numpy.array(data[k])
    arr = arr[:, [4, 5, 7]]
    if softmax:
      arr = numpy.exp(arr) / numpy.exp(arr).sum(axis=1)[:, None]
    else:
      arr = arr / arr.sum(axis=1)[:, None]
    for i in range(arr.shape[1]):
      axes[k].plot(arr[:, i])
    axes[k].legend(['X', 'R', 'Y'])
    if fixed_axis:
      axes[k].set_xlim(0, 5000)
      axes[k].set_ylim(0, 1.0)
    axes[k].set_xlabel('alpha_{}'.format(k))