valid.py 文件源码

def predict(self):
        """
        Train the regression model with predictions on validation set.
        Save the learned weights to apply to test set predictions.
        """
        pred_array = np.stack(self.pred_list, -1)
        reg = linear_model.Ridge(alpha=.5)
        pred = np.reshape(pred_array, [-1, len(self.pred_list)])
        y = np.reshape(self.labels_val, [-1,1])
        reg.fit(pred, y)

        self.weights = reg.coef_[0].tolist()