Created using Colaboratory

This commit is contained in:
udlbook
2023-12-24 11:49:04 -05:00
parent 0748270a1c
commit b9ec83d5f5

View File

@@ -4,7 +4,6 @@
"metadata": { "metadata": {
"colab": { "colab": {
"provenance": [], "provenance": [],
"authorship_tag": "ABX9TyPz1B8kFc21JvGTDwqniloA",
"include_colab_link": true "include_colab_link": true
}, },
"kernelspec": { "kernelspec": {
@@ -185,10 +184,8 @@
" if A[i,j] < 0:\n", " if A[i,j] < 0:\n",
" A[i,j] = 0;\n", " A[i,j] = 0;\n",
"\n", "\n",
" ATA = np.matmul(np.transpose(A), A)\n", " beta_omega = np.linalg.lstsq(A, y, rcond=None)[0]\n",
" ATAInv = np.linalg.inv(ATA)\n", "\n",
" ATAInvAT = np.matmul(ATAInv, np.transpose(A))\n",
" beta_omega = np.matmul(ATAInvAT,y)\n",
" beta = beta_omega[0]\n", " beta = beta_omega[0]\n",
" omega = beta_omega[1:]\n", " omega = beta_omega[1:]\n",
"\n", "\n",