岭回归中单位矩阵的形状

Question

我正在尝试为python中的b系数手动实现岭回归。我想根据以下公式制作逆矩阵：（X ^T X +λ I ） ^-1 ，但我不确定是哪个应该是我的身分矩阵 I 的形状。下面是我的代码。应该是x.shape [1]还是y.shape [1]

import statsmodels.api as sm
import numpy as np

x = np.random.randint(100, size = 20).reshape(10, 2) 
y = np.random.randint(100, size = 10).reshape(10, 1)
x = sm.add_constant(x)
alpha = 2

matrix_inverse = np.linalg.inv(x.T @ x + np.identity(x.shape[1]) * alpha)

是x.shape [1]还是我错了？