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):
- 创建类中的实例对象,注意前后都是两个下划线