Update params in inspector + docs

juaml · Oct 17, 2024 · d290cfe · d290cfe
1 parent b68a457
commit d290cfe
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/examples/99_docs/run_model_inspection_docs.py b/examples/99_docs/run_model_inspection_docs.py
@@ -140,7 +140,7 @@
 for fold_inspector in inspector.folds:
     fold_model = fold_inspector.model
     c_values.append(
-        fold_model.get_fitted_params()["svm__model_"].get_params()["C"]
+        fold_model.get_fitted_params()["svm__C"]
     )
 
 ##############################################################################

diff --git a/julearn/inspect/_pipeline.py b/julearn/inspect/_pipeline.py
@@ -63,11 +63,14 @@ def get_fitted_params(self):
                 ),
             }
 
-        return {
+        private_params = {
             param: val
             for param, val in all_params.items()
             if re.match(r"^[a-zA-Z].*[a-zA-Z0-9]*_$", param)
         }
+        out = self.get_params()
+        out.update(private_params)
+        return out
 
     @property
     def estimator(self):