MilesCranmer commited on
Commit
14e9a4b
1 Parent(s): c200074

Change default test kwargs to accurate model selection

Browse files
Files changed (1) hide show
  1. test/test.py +4 -1
test/test.py CHANGED
@@ -16,7 +16,10 @@ class TestPipeline(unittest.TestCase):
16
  default_niterations = (
17
  inspect.signature(PySRRegressor.__init__).parameters["niterations"].default
18
  )
19
- self.default_test_kwargs = dict(niterations=default_niterations * 2)
 
 
 
20
  np.random.seed(0)
21
  self.X = np.random.randn(100, 5)
22
 
 
16
  default_niterations = (
17
  inspect.signature(PySRRegressor.__init__).parameters["niterations"].default
18
  )
19
+ self.default_test_kwargs = dict(
20
+ niterations=default_niterations * 2,
21
+ model_selection="accuracy",
22
+ )
23
  np.random.seed(0)
24
  self.X = np.random.randn(100, 5)
25