Kernel Density Estimation (KDE) implementation in pytorch or tensorflow

user_14881301 0
Viewed 745

I found an implementation of the Kernel density estimation in scikit-learn as:

from sklearn.neighbors import KernelDensity
kde = KernelDensity(bandwidth=1.0, kernel='gaussian')
kde.fit(x[:, None])
logprob = kde.score_samples(x_d[:, None])

The problem is I want to use Automatic differentiation to take the derivative of logprob w.r.t. x, so I need to use pytorch or tensorflow.

Is there any implementation of the KDE in pytorch or tensorflow, so I can use AD afterward? Or, How can I calculate the derivative of logprob w.r.t. x with scikit-learn?

pytorch scikit-learn tensorflow python
1 Answers

JAX now has an implementation

Since you mentioned autograd, JAX is capable of both autograd and jit compilation and offers now the gaussian KDE function that also SciPy implements.

Update

Just to mention, there are also converters that may work, i.e. tf2jax and the other way around.

user_43605570
Related