Bases: numerical_kernel_based_interdependence
Source code in tinybig/interdependence/numerical_kernel_interdependence.py
| class gaussian_rbf_kernel_interdependence(numerical_kernel_based_interdependence):
def __init__(self, name: str = 'gaussian_rbf_kernel_interdependence', sigma: float = 1.0, *args, **kwargs):
gaussian_rbf_kernel_func = functools.partial(gaussian_rbf_kernel, sigma=sigma)
super().__init__(kernel=gaussian_rbf_kernel_func, name=name, *args, **kwargs)
|