def fit(self, X, y): n_lambdas, intercept_, coef_, _, _, rsquared_, lambdas, _, jerr \ = elastic_net(X, y, self.rho, lambdas=[self.alpha]) # elastic_net will fire exception instead # assert jerr == 0 self.coef_ = coef_ self.intercept_ = intercept_ self.rsquared_ = rsquared_ return self
def elastic_net_path(X, y, rho, **kwargs): """return full path for ElasticNet""" n_lambdas, intercepts, coefs, _, _, _, lambdas, _, jerr \ = elastic_net(X, y, rho, **kwargs) return lambdas, coefs, intercepts