Spaces:
Sleeping
Sleeping
Commit
·
e56007b
1
Parent(s):
56cb021
Correct use of Xresampled
Browse files- test/test.py +1 -2
test/test.py
CHANGED
|
@@ -217,11 +217,10 @@ class TestPipeline(unittest.TestCase):
|
|
| 217 |
unary_operators=["cos"],
|
| 218 |
select_k_features=3,
|
| 219 |
early_stop_condition=1e-4, # Stop once most accurate equation is <1e-4 MSE
|
| 220 |
-
Xresampled=Xresampled,
|
| 221 |
maxsize=12,
|
| 222 |
**self.default_test_kwargs,
|
| 223 |
)
|
| 224 |
-
model.fit(X, y)
|
| 225 |
model.set_params(model_selection="accuracy")
|
| 226 |
model.predict(X)
|
| 227 |
self.assertLess(np.average((model.predict(X) - y) ** 2), 1e-4)
|
|
|
|
| 217 |
unary_operators=["cos"],
|
| 218 |
select_k_features=3,
|
| 219 |
early_stop_condition=1e-4, # Stop once most accurate equation is <1e-4 MSE
|
|
|
|
| 220 |
maxsize=12,
|
| 221 |
**self.default_test_kwargs,
|
| 222 |
)
|
| 223 |
+
model.fit(X, y, Xresampled=Xresampled)
|
| 224 |
model.set_params(model_selection="accuracy")
|
| 225 |
model.predict(X)
|
| 226 |
self.assertLess(np.average((model.predict(X) - y) ** 2), 1e-4)
|