Skip to content

Commit

Permalink
Merge pull request #184 from Squadula/master
Browse files Browse the repository at this point in the history
Replace ChaospyLinReg by SklearnLinReg and extend LinearRegression parent class
  • Loading branch information
krystophny authored Oct 23, 2024
2 parents 29f3774 + 3388856 commit 1ea708c
Show file tree
Hide file tree
Showing 8 changed files with 620 additions and 242 deletions.
22 changes: 16 additions & 6 deletions profit/defaults.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,12 +36,22 @@
}

fit_linear_regression = {
"surrogate": "ChaospyLinreg",
"model": "monomial",
"order": 2,
"model_kwargs": None,
"sigma_n": 0.1,
"sigma_p": 10,
"surrogate": "SklearnLinreg",
"expansion": "legendre",
"expansion_kwargs": {}, # default for from_config
"poly_kwargs": {
"max_degree": 4, # default polynomial kwargs
"cross_truncation": 1.0,
"alpha": None,
"beta": None,
},
"rbf_kwargs": {
"rbf_type": "gaussian", # default rbf kwargs
"method": "grid",
"grid_size": 5,
"epsilon": 1.0,
},
"regressor": "BayesianRidge",
}

# Active Learning Config
Expand Down
2 changes: 1 addition & 1 deletion profit/sur/linreg/__init__.py
Original file line number Diff line number Diff line change
@@ -1,2 +1,2 @@
from .linear_regression import LinearRegression
from .chaospy_linreg import ChaospyLinReg
from .sklearn_linreg import SklearnLinReg
206 changes: 0 additions & 206 deletions profit/sur/linreg/chaospy_linreg.py

This file was deleted.

Loading

0 comments on commit 1ea708c

Please sign in to comment.