MilesCranmer commited on
Commit
af5ed6e
·
1 Parent(s): 9796d45

Turn optimization back on for testing

Browse files
Files changed (1) hide show
  1. test/test.py +0 -2
test/test.py CHANGED
@@ -5,7 +5,6 @@ X = np.random.randn(100, 5)
5
  # Test 1
6
  y = X[:, 0]
7
  equations = pysr(X, y,
8
- julia_optimization=0,
9
  niterations=100)
10
  print(equations)
11
  assert equations.iloc[-1]['MSE'] < 1e-10
@@ -14,7 +13,6 @@ assert equations.iloc[-1]['MSE'] < 1e-10
14
  y = X[:, 0]**2
15
  equations = pysr(X, y,
16
  unary_operators=["square(x) = x^2"], binary_operators=["plus"],
17
- julia_optimization=0,
18
  niterations=100)
19
  print(equations)
20
  assert equations.iloc[-1]['MSE'] < 1e-10
 
5
  # Test 1
6
  y = X[:, 0]
7
  equations = pysr(X, y,
 
8
  niterations=100)
9
  print(equations)
10
  assert equations.iloc[-1]['MSE'] < 1e-10
 
13
  y = X[:, 0]**2
14
  equations = pysr(X, y,
15
  unary_operators=["square(x) = x^2"], binary_operators=["plus"],
 
16
  niterations=100)
17
  print(equations)
18
  assert equations.iloc[-1]['MSE'] < 1e-10