From dd58f602ecafb4caa5dea7a5709c2d645f57ae6b Mon Sep 17 00:00:00 2001 From: jajupmochi Date: Thu, 2 Dec 2021 16:29:57 +0100 Subject: [PATCH] [Feature] Add kronecker_delta_kernel function between a pair of numpy vectors. --- gklearn/utils/kernels.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/gklearn/utils/kernels.py b/gklearn/utils/kernels.py index c500097..c35cc2f 100644 --- a/gklearn/utils/kernels.py +++ b/gklearn/utils/kernels.py @@ -4,7 +4,7 @@ These kernels are defined between pairs of vectors. import numpy as np -def delta_kernel(x, y): +def kronecker_delta_kernel(x, y): """Delta kernel. Return 1 if x == y, 0 otherwise. Parameters @@ -23,6 +23,10 @@ def delta_kernel(x, y): labeled graphs. In Proceedings of the 20th International Conference on Machine Learning, Washington, DC, United States, 2003. """ + return (1 if np.array_equal(x, y) else 0) + + +def delta_kernel(x, y): return x == y #(1 if condition else 0)