Browse Source

fix bugs in tests.

v0.1
jajupmochi 5 years ago
parent
commit
e4ae738487
2 changed files with 3 additions and 2 deletions
  1. +2
    -2
      gklearn/kernels/marginalizedKernel.py
  2. +1
    -0
      gklearn/kernels/randomWalkKernel.py

+ 2
- 2
gklearn/kernels/marginalizedKernel.py View File

@@ -259,8 +259,8 @@ def _marginalizedkernel_do(g1, g2, node_label, edge_label, p_quit, n_iteration):
for neighbor2 in neighbor_n2: for neighbor2 in neighbor_n2:
t_dict[(node1, node2, neighbor1, neighbor2)] = \ t_dict[(node1, node2, neighbor1, neighbor2)] = \
p_trans_n1 * p_trans_n2 * \ p_trans_n1 * p_trans_n2 * \
deltakernel(g1.node[neighbor1][node_label],
g2.node[neighbor2][node_label]) * \
deltakernel(g1.nodes[neighbor1][node_label],
g2.nodes[neighbor2][node_label]) * \
deltakernel( deltakernel(
neighbor_n1[neighbor1][edge_label], neighbor_n1[neighbor1][edge_label],
neighbor_n2[neighbor2][edge_label]) neighbor_n2[neighbor2][edge_label])


+ 1
- 0
gklearn/kernels/randomWalkKernel.py View File

@@ -9,6 +9,7 @@
import time import time
from functools import partial from functools import partial
from tqdm import tqdm from tqdm import tqdm
import sys


import networkx as nx import networkx as nx
import numpy as np import numpy as np


Loading…
Cancel
Save