From a3061244a6544ecf4e20087e0f02e8e36d47a6a8 Mon Sep 17 00:00:00 2001 From: jajupmochi Date: Fri, 2 Oct 2020 14:58:27 +0200 Subject: [PATCH] Fix a bug in weisfeilerLehmankernel. --- gklearn/kernels/weisfeilerLehmanKernel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gklearn/kernels/weisfeilerLehmanKernel.py b/gklearn/kernels/weisfeilerLehmanKernel.py index 960d391..222f5c5 100644 --- a/gklearn/kernels/weisfeilerLehmanKernel.py +++ b/gklearn/kernels/weisfeilerLehmanKernel.py @@ -256,7 +256,7 @@ def _wl_kernel_do(Gn, node_label, edge_label, height, parallel, n_jobs, chunksiz all_num_of_each_label.append(dict(Counter(labels_comp))) # calculate subtree kernel with h iterations and add it to the final kernel - compute_kernel_matrix(Kmatrix, all_num_of_each_label, Gn, parallel, n_jobs, False) + compute_kernel_matrix(Kmatrix, all_num_of_each_label, Gn, parallel, n_jobs, chunksize, False) return Kmatrix