PySRSequenceRegressor

.gitignore

@@ -25,3 +25,4 @@ site
 venv
 requirements-dev.lock
 requirements.lock
+.eggs/

docs/examples_sequence.md

@@ -0,0 +1,52 @@
+# Toy Sequence Examples with Code
+
+## Preamble
+```python
+import numpy as np
+from pysr import PySRSequenceRegressor
+```
+
+Note that most of the functionality
+of PySRSequenceRegressor is inherited
+from PySRRegressor.
+
+## 1. Simple Search
+
+Here's a simple example where we
+find the expression `f(n) = f(n-1) + f(n-2)`.
+
+```python
+X = [1, 1]
+for i in range(20):
+    X.append(X[-1] + X[-2])
+X = np.array(X)
+model = PySRSequenceRegressor(
+    recursive_history_length=2,
+    binary_operators=["+", "-", "*", "/"]
+)
+model.fit(X)  # no y needed
+print(model)
+```
+
+## 2. Multidimensionality
+
+Here we find a 2D recurrence relation
+with two data points at a time.
+
+```python
+X = [[1, 2], [3, 4]]
+for i in range(100):
+    X.append([
+        X[-1][0] + X[-2][0],
+        X[-1][1] - X[-2][1]
+    ])
+X = np.array(X)
+
+model = PySRSequenceRegressor(
+    recursive_history_length=2,
+    binary_operators=["+", "*"],
+)
+
+model.fit(X)
+print(model)
+```

environment.yml

@@ -5,7 +5,7 @@ dependencies:
   - python>=3.8
   - sympy>=1.0.0,<2.0.0
   - pandas>=0.21.0,<3.0.0
-  - numpy>=1.13.0,<2.0.0
+  - numpy>=1.20.0,<2.0.0
   - scikit-learn>=1.0.0,<2.0.0
   - pyjuliacall>=0.9.21,<0.9.22
   - click>=7.0.0,<9.0.0

pysr/__init__.py

@@ -8,6 +8,7 @@
 from .export_jax import sympy2jax
 from .export_torch import sympy2torch
 from .julia_extensions import load_all_packages
+from .regressor_sequence import PySRSequenceRegressor
 from .sr import PySRRegressor
 
 # This file is created by setuptools_scm during the build process:
@@ -22,6 +23,7 @@
     "install",
     "load_all_packages",
     "PySRRegressor",
+    "PySRSequenceRegressor",
     "best",
     "best_callable",
     "best_row",

pysr/export_latex.py

@@ -76,6 +76,9 @@ def sympy2latextable(
     if indices is None:
         indices = list(equations.index)
 
+    if output_variable_name == None:
+        output_variable_name = "y"
+
     for i in indices:
         latex_equation = sympy2latex(
             equations.iloc[i]["sympy_format"],