Spaces:
Running
Running
File size: 1,001 Bytes
bed9614 41e5fd5 05cf610 bed9614 10ff16a 48b723d 8cfda07 10ff16a bed9614 022de02 ddb4d52 48b723d ddb4d52 b078c6a e394c0e 10ff16a ddb4d52 022de02 8cfda07 48b723d 10ff16a 48b723d 022de02 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 |
import numpy as np
from pysr import pysr
import sympy
X = np.random.randn(100, 5)
default_test_kwargs = dict(
niterations=10,
populations=4,
user_input=False,
annealing=True,
useFrequency=False,
)
print("Test 1 - defaults; simple linear relation")
y = X[:, 0]
equations = pysr(X, y, **default_test_kwargs)
print(equations)
assert equations.iloc[-1]['MSE'] < 1e-4
print("Test 2 - test custom operator")
y = X[:, 0]**2
equations = pysr(X, y,
unary_operators=["sq(x) = x^2"], binary_operators=["plus"],
extra_sympy_mappings={'square': lambda x: x**2},
**default_test_kwargs)
print(equations)
assert equations.iloc[-1]['MSE'] < 1e-4
X = np.random.randn(100, 1)
y = X[:, 0] + 3.0
print("Test 3 - empty operator list, and single dimension input")
equations = pysr(X, y,
unary_operators=[], binary_operators=["plus"],
**default_test_kwargs)
print(equations)
assert equations.iloc[-1]['MSE'] < 1e-4
|