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]