Spaces:
Running
Running
MilesCranmer
commited on
Commit
·
af5ed6e
1
Parent(s):
9796d45
Turn optimization back on for testing
Browse files- 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
|