1. 学ぶ
    2. /
  2. コース
    3. /
  3. Pythonで学ぶ線形モデリング入門
Connected

演習

RSS の極小値を可視化する

この演習では、モデルパラメータの値を変えたときに RSS がどのように変化するかを計算し、可視化します。まず切片は固定し、傾きだけを変えます。各傾きに対してモデル値を計算し、その結果としての RSS を求めます。RSS の配列が得られたら、コードで最小の RSS を特定し、その最小値を与えた傾きを求めます。

あらかじめ読み込まれているデータ配列 x_data、y_data と、空のコンテナ rss_list を使って始めてください。

指示

100 XP
  • a1_array の各試行値 a1 について、model() でモデル値を予測し、続いて compute_rss() に y_data, y_model を渡して計算した rss_value を rss_list に保存します。
  • rss_list を np.array() に変換し、np.min() を使って rss_array の最小値を求めます。
  • np.where() を使って対応する試行値を見つけ、結果を best_a1 に代入します。
  • plot_rss_vs_parameters() を使って、図と一致しているかを視覚的に確認します。