python 类

class OLR:
    # 构造函数用于传输数据,将用于建模的 X、Y 传递进来,记录为自己的属性
    # 创建实例对象
    def __init__(self,X,Y):
        self.X = X
        self.Y = Y

        # 根据公式求解回归系数 A
    def fit(self):
        Xt = self.X.T
        XtX = np.dot(Xt,self.X)
        XtX_inv = np.linalg.inv(XtX)
        temp = np.dot(XtX_inv,Xt)
        self.A = np.dot(temp,self.Y)

    # 看回归系数
    def getCoef(self):
        return self.A

    def predict(self,X):
        Y = np.dot(X,self.A)
        return Y
  • def __init__(self,X,Y):
  • 创建类中的实例对象,注意前后都是两个下划线
posted @ 2022-12-06 17:13  行云止  阅读(20)  评论(0)    收藏  举报