Update BandedTRF examples

Author: vinaysraghavan

examples/banded_ridge_TRF_fitting/plot_banded_trf_comparison.py

@@ -195,9 +195,9 @@
 
     # --- Right Plot: Banded TRF (Independent Alpha) ---
     # In the banded approach, we look at the R-path for the specific feature band
-    banded_path = banded_model.alpha_paths_[feat]
+    banded_path = banded_model.alpha_paths_[b_idx]
     # Calculate marginal improvement relative to previous bands' max R
-    prev_banded_r = 0 if b_idx == 0 else np.max(banded_model.alpha_paths_[feature_list[b_idx-1]])
+    prev_banded_r = 0 if b_idx == 0 else np.max(banded_model.alpha_paths_[b_idx-1])
     banded_delta_path = banded_path - prev_banded_r
 
     best_banded_alpha = banded_model.feature_alphas_[feat]

examples/banded_ridge_TRF_fitting/plot_banded_trf_optimization.py

@@ -102,10 +102,10 @@
     fig, axes = plt.subplots(1, 2, figsize=(12, 4), sharey=False)
     for i, (mdl, ord_list) in enumerate(zip([model1, model2], [order_1, order_2])):
         feat = ord_list[b_idx]
-        path = mdl.alpha_paths_[feat]
+        path = mdl.alpha_paths_[b_idx]
 
         # Calculate Delta R Path relative to the max R of the previous band
-        prev_r = 0 if b_idx == 0 else np.max(mdl.alpha_paths_[ord_list[b_idx-1]])
+        prev_r = 0 if b_idx == 0 else np.max(mdl.alpha_paths_[b_idx-1])
         delta_path = path - prev_r
 
         best_alpha = mdl.feature_alphas_[feat]