Renamed BayesianSparseRegression to SparseBayesianRegression

dynamicslab · Dec 13, 2023 · 66112bb · 66112bb
1 parent acab720
commit 66112bb
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/pysindy/optimizers/__init__.py b/pysindy/optimizers/__init__.py
@@ -41,4 +41,5 @@
     "FROLS",
     "SINDyPI",
     "MIOSR",
+    "SBR",
 ]
diff --git a/pysindy/optimizers/sbr.py b/pysindy/optimizers/sbr.py
@@ -38,7 +38,7 @@ def __init__(
 
     def _reduce(self, x, y):
         # set up a sparse regression and sample.
-        regression = BayesianSparseRegression(self.tau_0, self.nu, self.s, self.lamb)
+        regression = SparseBayesianRegression(self.tau_0, self.nu, self.s, self.lamb)
         self.mcmc = regression.fit(x, y, **self.mcmc_kwargs)
 
         # get the variable names and the mean values from the samples.
@@ -55,7 +55,7 @@ def _reduce(self, x, y):
         self.coef_ = beta
 
 
-class BayesianSparseRegression:
+class SparseBayesianRegression:
     def __init__(self, tau_0=0.1, nu=4, s=2, lamb=1):
         # set hyperparameters
         self.tau_0 = tau_0