class LinearRegression: def fit(self, train_features, train_target): self.w = np.zeros(train_features.shape) self.w0 = pd.Series(data=[train_target.mean()]*train_target.shape[0]) def predict(self, test_features): return test_features * self.w + self.w0