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:
t_dict[(node1, node2, neighbor1, neighbor2)] = \
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(
neighbor_n1[neighbor1][edge_label],
neighbor_n2[neighbor2][edge_label])


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

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

import networkx as nx
import numpy as np


Loading…
Cancel
Save