R言語

【R言語】関数lmによる線形回帰 単回帰分析・重回帰分析 回帰分析で予測する

  1. HOME >
  2. R言語 >

【R言語】関数lmによる線形回帰 単回帰分析・重回帰分析 回帰分析で予測する

スポンサーリンク

R言語の回帰分析の実行方法を紹介していきます。回帰分析の実行方法だけでなく、回帰分析により作ったモデルでテストデータの予測も行っていきます。

実行方法では、回帰分析の結果の見方やその抽出方法を詳しく解説し、テストデータの予測などはプロットを用いて視覚的に分かりやすく解説していきます。

今回扱うプログラミングコードも全て、以下からダウンロード可能です。

テンプレートとして自由に扱ってください。

Rでデータ解析を始めるならコレ
¥3,300 (2022/06/01 17:20時点 | Amazon調べ)

単回帰分析

データセットの用意

今回は、次のrockという油田の石に関するデータセットを回帰分析に用います。

rockはnumeric型area、peri、shapefactor型のpermという変数で構成されています。

このデータセットを関数plotで描画すると、次のように各変数の相関図をプロットをすることができます。

相関図

また、この相関の尺度である相関係数は、相関行列を求める関数cov2corと、関数varを用いることで次のように列挙することが可能です。

補足として、分散を計算する関数varの引数にデータフレームなどの複数の変数から成るデータを代入すると、共分散行列を計算することが可能です。この共分散行列をcov2corの引数に代入することにより、共分散行列を対応する相関行列に変換することができます。

上の各相関係数(corMatrixの要素)を見ると、areaとperiの相関係数が0.8225064と高く、相関が強いことが分かります。

このように、回帰分析などの統計解析をおこなう前に、データをプロットしたり変数間の相関係数を求めたりすると、どのような解析を行えば良いのかが明確になります。

今回は、areaperiの相関が強いので、この変数に関する回帰分析を行っていきます。

実行方法

次のように、関数lm最初の引数formula2つ目の引数データセットを代入することで、簡単に単回帰分析を実行できます。

上の例では、periをareaで回帰しています。

これを統計学のモデルで表すと次のようになります。\begin{align}y_i = \beta_0 + \beta_1 x_i + \varepsilon_i, \ \ i = 1, \ldots, n,\end{align}ここに、\(y_i\)はperiを表す確率変数であり、\(y_i\sim N(\beta0 + \beta_1x_1, \sigma^2),\ \ \varepsilon_i \sim N(0, \sigma^2),\ i = 1, \ldots, n\)。

つまり、\(y_i\ (peri)\)と\(\beta_0 + \beta_1x_i\ (area),\ i = 1, \ldots, n\)との差が最小になるような\(\beta_0\)と\(\beta_1\)を計算しています。

最小二乗解である\(\beta_0\)と\(\beta_1\)については、プロットによる解釈で説明しています。

関数lmで得られた変数regressionの中身をみていきます。regressionを参照すると次のモデルと回帰係数が表示されます。

さらに詳細な結果が欲しい場合、関数summaryを用います。

上で与えた変数summaryRegを参照してみましょう。次のようにモデルや回帰係数に加え、残差(Std. Error)、t値(t value)、p値(Pr(>|t|))などの重要な値があることが分かります。

さらにCoefficientsの項目の変数areaの行に「***」の記号があります。

その下のSignif. codesにあるように、「***」p値が0.001未満であることが分かります。したがって、変数areaは有意(\beta_1 = 0でない)ことがいえます。

早速、これらの値を取りだしていきます。各検定結果はリストの要素として格納されているため、次のようにsummaryRegの後ろに$coefficientsを付けくわえることで、抽出可能です。

上の検定結果testResultsはデータフレームであるので、各列名を与えることで回帰係数やp値の列を取り出すことができます。

また、補足として有意な変数を取得したい場合、次のようにp値が有意水準(0.05)未満である行名を得ます。

また、目的変数\(Y_i\)に対する説明変数\(x_i\)の情報量の尺度である決定係数は次のようにして得られます。

決定係数R^2 = 0.6765168でるため、「モデルの当てはまりはまあまあである」ことが言えます。

解析結果の出力

解析結果をcsvファイルに出力してみましょう。

実行方法で得た検定結果をデータフレームにまとめ、write.csvを用いてcsvファイルに出力していきます。

回帰係数、残差、t値、p値、決定係数、修正済み決定係数を解析結果として出力してみます。

次のように関数cbindを用いて、testResultsと決定係数の列Rsquaredと修正済み決定係数の列adjRsquaredを結合させます。

testResultsを参照すると、次の解析結果の表が表示されます。

最後に、write.csvを用いて、このデータフレームtestResultsを出力します。

作業ディレクトリに単回帰分析.csvが出力され、次の画像のように単回帰分析の結果が保存されていることが確認できます。

単回帰分析結果

テストデータを用いた予測

回帰分析のモデルの使用例として、テストデータ等の予測があります。

モデルを作ることによって、未知のデータを予測できるようになります。簡単な実行例をみていきましょう。

テストデータとして、次のデータセットvalidationDataを用います。このデータは、さきほどのデータセットrockに正規乱数を加えたものとなります。

validationDataを参照すると、次のような乱数が加わったデータであることが確認できます。

実行方法で得たモデルregressionに、このテストデータの数値をあてはめたい場合、関数predictを用います。

最初の引数lmで得られたオブジェクト2つ目あてはめたいデータを代入することで予測することが可能です。

また、次のように予測結果をプロットすることができます。

プロット画面に次のような回帰直線が描画されます。青線がテストデータをモデルにあてはめた際の値を通る回帰直線で、黒い点がテストデータの実際の値になります。テストデータであるperiをある程度予測できていることが確認できました。

テストデータあてはめ

プロットによる解析結果の解釈

単回帰分析の幾何学的解釈をしていきます。

まず、install.packagesggplot2をインストールしましょう。このパッケージを用いることで各種統計解析の結果を簡単にプロットすることができます。

インストールしライブラリに追加したら、次のように関数ggplotを実行しましょう。

変数periとareaのデータのプロット及び回帰直線とその信頼区間の描画をすることができます。上を実行すると次の画像がプロットされます。

単回帰分析プロット

dataset$areaの各値\(x_i\)を\(y_i=\beta_0 + \beta_1 x_i = -471.4357929+ 0.4387544\times x_i ,\  i= 1,\ldots, n\)にあてはめた値を通る直線が上の赤い直線です。

また、濃いグレーの領域は\(y_i\)の95%信頼区間を表します。

重回帰分析

単回帰分析に続いて、R言語の重回帰分析の実行方法や解析結果の見方を紹介していきます。

単回帰分析のデータセットの用意で用いたデータセットrockを今回も使用します。

重回帰分析により、peri以外の全ての変数でperiを予測することが可能となります。

単回帰分析ではperiをareaで予測しましたが、他の変数を取り入れることでareaだけでは説明できない情報が加わり、予測精度が向上することができます(必ずしも向上しません)。

実行方法

重回帰分析は関数lmを用いることで実行できます。lm2つ目の引数datadatasetを代入し、1つ目の引数のformulaperi ~.を代入することで、peri以外の変数を説明変数とする重回帰分析を実行できます。

これは、次の統計学のモデルを意味します。\begin{align}Y_i = \beta_0 + \beta_1x_{i1} + \beta_2x_{i2} + \cdots + \beta_px_{ip} + \varepsilon_i = \boldsymbol{x}_i^T\boldsymbol{\beta},\  \ i = 1, \ldots, n,\end{align}ここに\begin{align}\boldsymbol{x}_i &= \begin{pmatrix}1\\x_{i1}\\ \vdots \\ x_{ip}\end{pmatrix}\ \ \ \ \boldsymbol{\beta} = \begin{pmatrix}\beta_0\\\beta_1\\\vdots\\\beta_p\end{pmatrix}\end{align}であり、\(Y_i\sim N(\boldsymbol{x}_i^T\boldsymbol{\beta}, \sigma^2),\ \  \varepsilon_i \sim N(0, \sigma^2)\)である。

lmの第1引数のformulaを書き換えることで、いろいろなモデルを構築することが可能です。

すなわち、次の実行結果は上と同じです。

また、+演算子を用いることで次のように任意の説明変数を選択することも可能です。

これは、areashapeの2変数を説明変数とした重回帰分析となります。

また、補足として交互作用ありのモデルは次のように表現できます。

*演算子用いることで変数間の交互作用を含めることができます。

関数summaryを用いることで、解析結果得ることができます。

上の検定結果の表(Coefficients)の項目を見ると、(Intercept)areaperm「**、***」の記号がついているのが分かります。

その下のSignif. codesの項目を見ると、「0 ‘***’ 0.001 ‘**’ 0.01 ‘*’ 0.05 ‘.’ 0.1 ‘ ’ 1」書いており、「**」p値が0.01未満であり、「***」0.001未満であることが分かります。

よって、(Intercept)とareaとpermは有意である(\beta_0 = 0、\beta_2 = 0、\beta_3 = 0でない)ことがいえます。

上の解析結果を取り出すには、単回帰分析と同様にリストsummaryRegの各要素にアクセスする必要があります。

次のようにsummaryRegの後ろに$coefficientsを付けることで、回帰係数、残差、t値、p値を取得することができます。

p値0.05未満の変数を取得したい場合、次のように"Pr(>|t|)"列が0.05未満であるtestResultsの行名を抽出します。

また、目的変数に対する説明変数の情報量の指標である重相関係数は次のようにして得られます。

上の結果から、重相関係数R^2 = 0.8815147であり、モデルの当てはまりが非常に良いことが言えます。

解析結果の出力

解析結果をcsvファイルへ出力する例を紹介します。

単回帰分析と同様に、次のようなCoefficientsの表と重相関係数からなるデータフレームを出力する方法を考えます。

データフレームtestResultsには、切片項と各説明変数に対する回帰係数などが含まれているのが分かります。

write.csvを用いて、"重回帰分析.csv"というファイルに出力してみましょう。

作業ディレクトリを確認すると、"重回帰分析.csv"が出力されているのが分かります。ファイルを開くと解析結果が保存されていることがわかります。

重回帰分析結果

テストデータを用いた予測

モデリングが終わったので、テストデータを先ほどのモデルにあてはめてみましょう。

テストデータを用いた予測で用いたテストデータvalidationDataを再び用います。

validationDataを先ほどのモデルにあてはめたい場合、関数predictを用います。

predict1つ目の引数関数lmで得られたオブジェクト2つ目の引数モデルにあてはめたいデータセット(テストデータ)を代入することで、テストデータの予測が可能です。

上のようにhead(predictData)を参照すると、1行目から6のテストデータをモデルにあてはめた結果が返ってきます。

次にこれらのテストデータとその予測値をプロットして、予測の精度を可視化させましょう。

次を実行することで、テストデータ(黒まる)と予測値(青ばつ)のプロットが可能です。次のような画像がプロットされます。

重回帰分析テストデータ予測

x軸は標本のインデックス(行番号)であり、だいたい同じ領域に予測値が描画されているのが分かります。

プロットによる解析結果の解釈(説明変数2個)

最後に、重回帰分析の解析結果(回帰平面)をプロットで可視化させます。

単回帰分析の時は、回帰直線でモデルの予測値を描画することができましたが、説明変数が2個以上になると解釈が難しくなります。

今回は、例として説明変数が2個の場合を考えます。次のareaとshapeを説明変数とするモデルを考えます。

次にプロットの準備として、次のパッケージをインストールします。

パッケージscatterplot3dをインストールすることで、3Dの図形を描画することが可能となります。

パッケージをインストールしたら、グラフのx、z軸をそれぞれarea、shapeに設定し、グラフのy軸予測値(関数predictの値)に設定することで、回帰平面と実際のデータの座標を描画することができます。

上を実行することで、次のような画像がプロットされます。

回帰平面実行方法で紹介した統計モデルで、目的変数\(Y_i, \ i = 1, \ldots, n\)はp次元空間上のベクトル\(\boldsymbol{x}_i^T\boldsymbol{\beta}\)であることを仮定しました。

この例では、\(Y_i\)は2次元空間上のベクトルで与えられるため、平面上に分布することが分かります(\(\beta_0 = \beta_1 = \beta_2 = 0\)のとき、\(Y_i=0\)の平面)。

まとめ

R言語で回帰分析の実行方法や解析結果の見方や出力の方法をみていきました。

また、応用としてテストデータを用いた予測や回帰分析のプロットなどを行いました。

次回は、一般化線形モデルの実行方法について解説していきます。

代表的な一般化線形回帰であるロジスティック回帰分析の方法を紹介していきます。

スポンサーリンク

  • この記事を書いた人
  • 最新記事

usagi-san

統計学とゲームとかをメインに解説していくよ。 数式とかプログラミングコードにミスがあったり質問があったりする場合はコメントで受け付けます。すぐに対応します。

-R言語
-, , ,