From 6e2b2fa164a5dd0622f015432836b2f0aed0cbe1 Mon Sep 17 00:00:00 2001 From: jajupmochi Date: Sat, 26 Dec 2020 13:23:37 +0100 Subject: [PATCH] Fix a bug in WL subtree kernel. --- gklearn/kernels/weisfeiler_lehman.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gklearn/kernels/weisfeiler_lehman.py b/gklearn/kernels/weisfeiler_lehman.py index da14ff8..9fb7c83 100644 --- a/gklearn/kernels/weisfeiler_lehman.py +++ b/gklearn/kernels/weisfeiler_lehman.py @@ -231,7 +231,7 @@ class WeisfeilerLehman(GraphKernel): # @todo: sp, edge user kernel. return i, j, self.pairwise_kernel(G_gn[i], G_gn[j]) - def _compute_kernel_itr(kernel, all_num_of_each_label): + def _compute_kernel_itr(self, kernel, all_num_of_each_label): labels = set(list(all_num_of_each_label[0].keys()) + list(all_num_of_each_label[1].keys())) vector1 = np.array([(all_num_of_each_label[0][label]