Spaces:
Sleeping
Sleeping
File size: 506 Bytes
bed9614 8cfda07 ddb4d52 8cfda07 ddb4d52 bed9614 ddb4d52 8cfda07 ddb4d52 8cfda07 ddb4d52 8cfda07 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
import numpy as np
from pysr import pysr
X = np.random.randn(100, 5)
# Test 1
y = X[:, 0]
equations = pysr(X, y,
julia_optimization=0,
niterations=100)
print(equations)
assert equations.iloc[-1]['MSE'] < 1e-10
# Test 2
y = X[:, 0]**2
equations = pysr(X, y,
unary_operators=["square(x) = x^2"], binary_operators=["plus"],
julia_optimization=0,
niterations=100)
print(equations)
assert equations.iloc[-1]['MSE'] < 1e-10
|