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)
|