Spaces:
Running
Running
MilesCranmer
commited on
Commit
•
4582e28
1
Parent(s):
58834e8
Add missing weights argument to fit()
Browse files- pysr/sklearn.py +2 -1
pysr/sklearn.py
CHANGED
@@ -42,10 +42,11 @@ class PySRRegressor(BaseEstimator):
|
|
42 |
else:
|
43 |
raise NotImplementedError
|
44 |
|
45 |
-
def fit(self, X, y):
|
46 |
self.equations = pysr(
|
47 |
X=X,
|
48 |
y=y,
|
|
|
49 |
**self.params,
|
50 |
)
|
51 |
return self
|
|
|
42 |
else:
|
43 |
raise NotImplementedError
|
44 |
|
45 |
+
def fit(self, X, y, weights=None):
|
46 |
self.equations = pysr(
|
47 |
X=X,
|
48 |
y=y,
|
49 |
+
weights=weights,
|
50 |
**self.params,
|
51 |
)
|
52 |
return self
|