Skip to content

Instantly share code, notes, and snippets.

@mustafa-qamaruddin
Created May 4, 2019 17:26
Show Gist options
  • Save mustafa-qamaruddin/7b19c7a6f4228d2e3041e17a247440d4 to your computer and use it in GitHub Desktop.
Save mustafa-qamaruddin/7b19c7a6f4228d2e3041e17a247440d4 to your computer and use it in GitHub Desktop.
def build_model(_alpha, _l1_ratio):
estimator = ElasticNet(
alpha=_alpha,
l1_ratio=_l1_ratio,
fit_intercept=True,
normalize=False,
precompute=False,
max_iter=16,
copy_X=True,
tol=0.1,
warm_start=False,
positive=False,
random_state=None,
selection='random'
)
return MultiOutputRegressor(estimator, n_jobs=4)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment