Применение ядер для разделения измерений в gpytorch

Я перевожу трехмерный код регрессии гауссовского процесса из GPflow в GPyTorch, чтобы воспользоваться преимуществами встроенной масштабируемости PyTorch для точного георадара.

Этот пример полезен, но не может найти объяснения, как применять отдельные ядра к отдельным измерениям в многомерном георадаре (например, ядро ​​Matern52 для измерений [0,1] и ядро ​​Matern 32 для измерения [2]).

Поддерживает ли GPyTorch эту функциональность, и если да, то как она реализована / где я могу найти пример?

1 ответ

Решение

Используйте active_dims в функциях ядра:

from(https://docs.gpytorch.ai/en/v1.1.1/kernels.html) active_dims (кортеж целых чисел, необязательно): установите это, если вы хотите вычислить ковариацию только нескольких входных измерений. Цифры соответствуют индексам размеров. По умолчанию: Нет.

Другие вопросы по тегам