Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Replace ChaospyLinReg by SklearnLinReg and extend LinearRegression parent class #184

Merged
merged 2 commits into from
Oct 23, 2024

Conversation

Squadula
Copy link
Contributor

@Squadula Squadula commented Oct 22, 2024

The model construction and the regression part are now handled at different layers to allow for easy extension of the linear regression models:

Model construction (basis function expansion)

The basis function expansion is taken care of in the LinearRegression parent class. Either predefined expansions (orthogonal polynomials or RBFs) or custom expansions can be set and generated. This allows to add child classes that focus only on the regression task.

(Bayesian) linear regression

The SklearnLinReg child class trains the models and makes prediction using scikit-learn's BayesianRidge and ARDRegression classes.

Squadula and others added 2 commits October 22, 2024 04:56
…r predefined or custom set basis expansions, adjusted defaults and integration test
@Squadula Squadula added the enhancement New feature or request label Oct 22, 2024
@Squadula Squadula requested a review from krystophny October 22, 2024 15:25
@krystophny krystophny merged commit 1ea708c into redmod-team:master Oct 23, 2024
6 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants