Merge pull request #70 from MatthewSZhang/narx-speed

FEAT add time-narx to evaluate opt speed of narx fit

Author: MatthewSZhang

pixi.lock

@@ -95,6 +95,7 @@ compilers = "*"
 time-h = "python -m timeit -n 5 -s 'import numpy as np; from fastcan import FastCan; X = np.random.rand(3000, 100); y = np.random.rand(3000, 20); s = FastCan(100, verbose=0).fit(X, y)'"
 time-eta = "python -m timeit -n 5 -s 'import numpy as np; from fastcan import FastCan; X = np.random.rand(3000, 100); y = np.random.rand(3000, 20); s = FastCan(100, eta=True, verbose=0).fit(X, y)'"
 profile-minibatch = { cmd = '''python -c "import cProfile; import numpy as np; from fastcan import minibatch; X = np.random.rand(100, 3000); y = np.random.rand(100, 20); cProfile.run('minibatch(X, y, 1000, 10, verbose=0)', sort='$SORT')"''', env = { SORT = "cumtime" } }
+time-narx = '''python -m timeit -n 1 -s "import numpy as np; from fastcan.narx import make_narx; rng = np.random.default_rng(5); X = rng.random((1000, 10)); y = rng.random((1000, 2)); s = make_narx(X, y, 10, max_delay=2, poly_degree=2, verbose=0).fit(X, y, coef_init='one_step_ahead', verbose=1)"'''
 
 [tool.pixi.feature.test.tasks]
 test = { cmd = "pytest", cwd = "tests" }