def plot_eigenvalues()

in plot_path_tools.py [0:0]

• 24 lines of code
• 8 McCabe index (conditional complexity)

def plot_eigenvalues(gen_eigs, dis_eigs, game_eigs, labels=None, out_dir=None, summary_writer=None, step=0):
    """
    plots interpolation path in `hist` and computed by `compute_path_stats`.
    """
    assert out_dir is not None or summary_writer is not None, 'save results either as files in out_dir or in tensorboard!'

    if not os.path.exists(out_dir):
        os.makedirs(out_dir)

    fig1 = plt.figure()
    for i, eigs in enumerate(game_eigs):
        plt.scatter(eigs.real, eigs.imag, label=labels[i])
    plt.legend()

    fig2 = plt.figure()
    for i, eigs in enumerate(gen_eigs):
        plt.bar(np.arange(len(eigs)), eigs[::-1], label=labels[i])
    plt.legend()

    fig3 = plt.figure()
    for i, eigs in enumerate(dis_eigs):
        plt.bar(np.arange(len(eigs)), eigs[::-1], label=labels[i])
    plt.legend()

    if out_dir is not None:
        fig1.savefig(os.path.join(out_dir, 'game_eigs_%06d.png' % step))
        fig2.savefig(os.path.join(out_dir, 'gen_eigs_%06d.png' % step))
        fig3.savefig(os.path.join(out_dir, 'dis_eigs_%06d.png' % step))

    if summary_writer is not None:
        summary_writer.add_figure('game_eigs', fig1, step)
        summary_writer.add_figure('gen_eigs', fig2, step)
        summary_writer.add_figure('dis_eigs', fig3, step)