Gaussian kernel using torch

transferwise / hisel

Feature selection tool based on Hilbert-Schmidt Independence Criterion

Apache License 2.0

2 stars 0 forks source link

Gaussian kernel using torch #1

Closed claudio-tw closed 1 year ago

claudio-tw commented 1 year ago

Context

We give a version of the module kernels that uses pytorch instead of numpy. This version is called torchkernels. It supports computations on GPU if CUDA is available.

Checklist

[ ] Change meets or does not compromise the Baseline Security Requirements