Skip to content

Commit ebfe5ce

Browse files
Add the possibility to give an array-like input for n_comp option in KPLS (#816)
… and test the range of n_dim and choose the best one. It has been refactored to take into account the advice given by @Paul-Saves. I added unit tests (some of them are quite long due to the exhaustive search, so I marked them as `@unittest.skipIf(int(os.getenv("RUN_SLOW_TESTS", 0)) < 1, "too slow")`). I updated the docs with a new example and added some description to the parameters. I also added a little safeguard on the number of k_fold min based on the number of training points: ```python k_fold = min(4, original_nt) ``` --------- Co-authored-by: Saves Paul <paul.saves@alumni.enac.fr>
1 parent ff09068 commit ebfe5ce

6 files changed

Lines changed: 496 additions & 80 deletions

File tree

doc/_src_docs/surrogate_models/gpr/kpls.rst

Lines changed: 106 additions & 15 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

doc/_src_docs/surrogate_models/gpr/kpls.rstx

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,11 @@ Usage with an automatic number of components
3131

3232
.. embed-test-print :: smt.surrogate_models.tests.test_surrogate_model_examples , Test , test_kpls_auto
3333

34+
Usage with a range of number of components
35+
--------------------------------------------
36+
37+
.. embed-test-print :: smt.surrogate_models.tests.test_surrogate_model_examples , Test , test_kpls_range
38+
3439
Options
3540
-------
3641

0 commit comments

Comments
 (0)