1. ホーム
  2. ディープラーニング

plt.fillbetweenが行うこと

2022-03-20 22:44:10

plt.fill_betweenは、2つの曲線の間の領域を色で塗りつぶします。



plt.fill_between(x, y1, y2) は、x を水平座標、y1 と y2 を二つの曲線としたとき、二つの曲線 y1 と y2 の間の領域を色で塗りつぶす関数である。

この関数の最も一般的な使い方は、次の例のように、平均曲線と標準偏差を知り、標準偏差で覆われる領域を描画することである。

回帰問題では、入力ベクトルを x、出力の平均と標準偏差を y_mean と y_std というベクトルで、複数回の試行を経て得られることが知られている。ここで、2倍分散と2倍分散でカバーされる領域を描画する必要があり、そのコードは以下の通りである。

for i in range(2):
    plt.fill_between(x,
        y_mean - y_std * ((i+1.)),
        y_mean + y_std * ((i+1.)),
        color="b",
        alpha=0.1)
plt.legend()
plt.grid()