gavindiaz/optimiz-rs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: updated 3 more notebooks for v1.0.0 API - 75% success rate

Browse Source
This commit is contained in:
Melvin Alvarez
2026-02-16 17:35:15 +01:00
parent c18788160a
commit 6ebd1337fa
3 changed files with 398 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/notebooks/04_kalman_filter_sensor_fusion.ipynb
+121 -23
View File
File diff suppressed because one or more lines are too long
examples/notebooks/04_real_world_applications.ipynb
+272 -39
View File
File diff suppressed because one or more lines are too long
examples/notebooks/05_performance_benchmarks.ipynb
+5 -7
View File
@@ -98,7 +98,7 @@
" # Benchmark OptimizR (Rust)\n", " # Benchmark OptimizR (Rust)\n",
" rust_times = []\n", " rust_times = []\n",
" for _ in range(n_runs):\n", " for _ in range(n_runs):\n",
" hmm_rust = HMM(n_states=3, random_state=42)\n", " hmm_rust = HMM(n_states=3)\n",
" start = time.perf_counter()\n", " start = time.perf_counter()\n",
" hmm_rust.fit(data, n_iterations=50, tolerance=1e-4)\n", " hmm_rust.fit(data, n_iterations=50, tolerance=1e-4)\n",
" rust_times.append(time.perf_counter() - start)\n", " rust_times.append(time.perf_counter() - start)\n",
@@ -226,12 +226,11 @@
" rust_times = []\n", " rust_times = []\n",
" for _ in range(n_runs):\n", " for _ in range(n_runs):\n",
" start = time.perf_counter()\n", " start = time.perf_counter()\n",
" samples_rust, _ = mcmc_sample(\n", " samples_rust = mcmc_sample(\n",
" log_likelihood_fn=log_likelihood_normal,\n", " log_likelihood_fn=lambda params: log_likelihood_normal(params, data),\n",
" data=data,\n",
" initial_params=[0.0, 0.05],\n", " initial_params=[0.0, 0.05],\n",
" param_bounds=[(-1.0, 1.0), (0.001, 1.0)],\n", " param_bounds=[(-1.0, 1.0), (0.001, 1.0)],\n",
" proposal_std=[0.01, 0.005],\n", " proposal_std=0.01,\n",
" n_samples=n_samples,\n", " n_samples=n_samples,\n",
" burn_in=1000\n", " burn_in=1000\n",
" )\n", " )\n",
@@ -245,8 +244,7 @@
" for _ in range(n_runs):\n", " for _ in range(n_runs):\n",
" start = time.perf_counter()\n", " start = time.perf_counter()\n",
" samples_py, _ = mcmc_python(\n", " samples_py, _ = mcmc_python(\n",
" log_likelihood_fn=log_likelihood_normal,\n", " log_likelihood_fn=lambda params: log_likelihood_normal(params, data),\n",
" data=data,\n",
" initial_params=[0.0, 0.05],\n", " initial_params=[0.0, 0.05],\n",
" param_bounds=[(-1.0, 1.0), (0.001, 1.0)],\n", " param_bounds=[(-1.0, 1.0), (0.001, 1.0)],\n",
" proposal_std=np.array([0.01, 0.005]),\n", " proposal_std=np.array([0.01, 0.005]),\n",