Merge pull request #54 from MatthewSZhang/multi-output

FEAT make NARX accept multi-output

Author: MatthewSZhang

.github/pull_request_template.md

@@ -0,0 +1,5 @@
+## Checklist
+
+- [ ] Used a personal fork to propose changes
+- [ ] A reference to a related issue:
+- [ ] A description of the changes

.github/workflows/ci.yml

@@ -22,7 +22,7 @@ jobs:
     steps:
       - uses: actions/checkout@v4
       - name: Build wheels
-        uses: pypa/cibuildwheel@v2.22.0
+        uses: pypa/cibuildwheel@v2.23.0
         env:
           CIBW_BUILD: cp3*-*
           CIBW_SKIP: pp* *i686* *musllinux* *-macosx_universal2 *-manylinux_ppc64le *-manylinux_s390x

examples/plot_narx.py

@@ -145,7 +145,7 @@
 auto_narx_model = make_narx(
     X=X,
     y=y,
-    n_features_to_select=4,
+    n_terms_to_select=4,
     max_delay=3,
     poly_degree=2,
     verbose=0,

examples/plot_narx_msa.py

@@ -110,7 +110,7 @@ def auto_duffing_equation(y, t):
 narx_model = make_narx(
     X=u_train,
     y=y_train,
-    n_features_to_select=10,
+    n_terms_to_select=10,
     max_delay=max_delay,
     poly_degree=3,
     verbose=0,
@@ -159,7 +159,7 @@ def plot_prediction(ax, t, y_true, y_pred, title):
 narx_model = make_narx(
     X=u_all,
     y=y_all,
-    n_features_to_select=10,
+    n_terms_to_select=10,
     max_delay=max_delay,
     poly_degree=3,
     verbose=0,