深層学習の層の数と精度変化の歴史のグラフ

7276 ワード
論文 発表 Python Python テキストリンク

深層学習の発展のグラフ

深層学習の発展を表すグラフとしてよくILSVRCにおける精度の変化と層の数の推移を表すグラフを見かけるので作っておきました．

pythonのコード

import matplotlib.pyplot as plt


def plot_res():
    plt.rcParams["font.family"] = "Arial"
    plt.rcParams["font.size"] = 18
    fig = plt.figure()

    c = ["skyblue", "blue"] * 5
    depths = [8, 8, 22, 152, 205]
    years = list(range(2012, 2017))
    errors = [16.4, 11.7, 7.3, 6.7, 2.9]
    a = 0.2

    ax1 = fig.add_subplot(111)
    depth_bars = ax1.bar(years, depths, color=c[0], tick_label=years)
    for year, depth in zip(years, depths):
        ax1.text(year, depth, depth if depth < 200 else "200>", ha="center", va="bottom", color="black", fontsize=14)
    ax1.set_ylim(0, 230)
    ax1.set_ylabel("Depth of Neural Networks")

    ax2 = ax1.twinx()
    error_line = ax2.plot(years, errors, color=c[1], linewidth=2)
    for year, error in zip(years, errors):
        ax2.text(year - a, error + 1, error, color="black", fontsize=14)
    ax2.set_ylim(0, 20)
    ax2.set_ylabel("Error Rate")

    plt.title("Winner in ILSVRC")
    plt.tight_layout()
    plt.show()

plot_res()