Kernel dimension control by AdrianSosic · Pull Request #748 · emdgroup/baybe

AdrianSosic · 2026-02-12T12:53:34Z

DevPR, parent is #745

Enables to specify a kernel's active dimensions in a user-friendly way, i.e. by passing parameter_names
Adds the functionality for parameter subselection
Absorbs the ICM construction for multi-task models from the GP code into a new ICMKernelFactory class

Use the proper aggregated variable instead of the loop-local

* Add validator * Forbid string to tuple conversion

This is potentially only a temporary solution until we find a better design or figure out additional requirements. The class is private to simplify future refactorings.

AdrianSosic self-assigned this Feb 12, 2026

AdrianSosic added the enhancement Expand / change existing functionality label Feb 12, 2026

AdrianSosic mentioned this pull request Feb 12, 2026

Beef up GaussianProcessSurrogate #745

Draft

20 tasks

AdrianSosic changed the base branch from main to dev/gp February 12, 2026 12:56

AdrianSosic added the dev label Feb 12, 2026

AdrianSosic force-pushed the refactor/multi_task branch from 69ebbfa to 0144c44 Compare February 12, 2026 14:05

AdrianSosic force-pushed the refactor/multi_task branch 2 times, most recently from 5cda8d9 to a12c11f Compare February 25, 2026 13:57

jpenn2023 reviewed Feb 25, 2026

View reviewed changes

Comment thread baybe/surrogates/gaussian_process/components/kernel.py Outdated

AdrianSosic force-pushed the refactor/multi_task branch from a12c11f to 7dbfde7 Compare February 26, 2026 09:25

jpenn2023 reviewed Feb 27, 2026

View reviewed changes

Comment thread baybe/surrogates/gaussian_process/components/__init__.py

jpenn2023 reviewed Feb 27, 2026

View reviewed changes

Comment thread baybe/surrogates/gaussian_process/presets/edbo.py Outdated

AdrianSosic force-pushed the refactor/multi_task branch 6 times, most recently from 0bc8e21 to 9d285ce Compare March 6, 2026 10:31

AdrianSosic added 5 commits March 6, 2026 17:54

Absorb index kernel construction into ICMKernelFactory

b84af1d

Add parameter selectors

e365862

Rename protocols

dc9ee90

Implement active kernel dimension control

63e64fa

Move parameter_names attribute down to BasicKernel subclass

da8ffc0

AdrianSosic force-pushed the refactor/multi_task branch from 9d285ce to ffffafe Compare March 6, 2026 17:03

jpenn2023 reviewed Mar 27, 2026

View reviewed changes

Comment thread baybe/surrogates/gaussian_process/core.py

jpenn2023 mentioned this pull request Mar 27, 2026

Multi-fidelity surrogate models #721

Open

AdrianSosic added 4 commits April 1, 2026 14:55

Fix condition in BayBEKernelFactory

86e110b

Add deprecation mechanism for breaking change in kernel logic

3714c5f

Import KernelFactory to components/__init__.py

66b3dfb

Add citation to docstring

d91623f

AdrianSosic added 5 commits April 10, 2026 09:52

Fix variable reference in kernel translation test

8480607

Use the proper aggregated variable instead of the loop-local

Use chain.from_iterable instead of unpacking

7cd2321

Replace hard-coded parameter type in deprecation error message

9cc4647

Add comment explaining the role of active_dims=None

9c5cf18

Fix keyword logic in Kernel.to_gpytorch

d13e365

Copilot started work on behalf of AdrianSosic April 10, 2026 08:48 View session

Copilot finished work on behalf of AdrianSosic April 10, 2026 08:54

AdrianSosic added 11 commits April 10, 2026 11:02

Add NameSelector class

1707e3f

Refine NameSelector.parameter_names field specification

c859d85

* Add validator * Forbid string to tuple conversion

Make KernelFactory class abstract

3a1c421

Remove unnecessary keyword assignment

66b6f06

Add converters to ICMKernelFactory

4edf552

Revise comment on kernel attribute matching

a1bc442

Fix converter specification

4811dae

Add convenience converter for parameter selectors

4347c76

Turn KernelFactory into mixin class for parameter selection

88468a1

This is potentially only a temporary solution until we find a better design or figure out additional requirements. The class is private to simplify future refactorings.

Override default selectors for existing factories

2d414a2

Turn assert statement into proper AssertionError

1b248a5

AdrianSosic requested a review from Scienfitz April 10, 2026 17:03

AdrianSosic added 3 commits April 13, 2026 14:00

Use regex by default for NameSelector

f18cd32

Rename parameter_types attribute of TypeSelector to types

a6d611e

Rephrase changelog entry

55582fc

Scienfitz approved these changes Apr 13, 2026

View reviewed changes

AdrianSosic added 2 commits April 13, 2026 15:57

Merge branch 'dev/gp' into refactor/multi_task

54a5674

Fix typing issues

20f106b

AdrianSosic force-pushed the refactor/multi_task branch from 59418eb to 20f106b Compare April 13, 2026 14:26

AdrianSosic added 2 commits April 13, 2026 20:08

Add ignore to docs/conf.py

3c38d34

Add missing searchspace argument

f943eeb

AdrianSosic merged commit 9ef2604 into dev/gp Apr 13, 2026
12 of 13 checks passed

AdrianSosic deleted the refactor/multi_task branch April 13, 2026 18:43

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Kernel dimension control#748

Kernel dimension control#748
AdrianSosic merged 44 commits intodev/gpfrom
refactor/multi_task

AdrianSosic commented Feb 12, 2026 •

edited

Loading

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

5 participants

Conversation

AdrianSosic commented Feb 12, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

5 participants

AdrianSosic commented Feb 12, 2026 •

edited

Loading