Browse Source

Fix GEDEnv import bugs.

v0.2.x
jajupmochi 5 years ago
parent
commit
58cc4042cc
1 changed files with 4 additions and 1 deletions
  1. +4
    -1
      gklearn/ged/util/util.py

+ 4
- 1
gklearn/ged/util/util.py View File

@@ -14,10 +14,11 @@ import sys
from tqdm import tqdm from tqdm import tqdm
import networkx as nx import networkx as nx
from gklearn.ged.env import GEDEnv from gklearn.ged.env import GEDEnv
from gklearn.gedlib import librariesImport, gedlibpy




def compute_ged(g1, g2, options): def compute_ged(g1, g2, options):
from gklearn.gedlib import librariesImport, gedlibpy

ged_env = gedlibpy.GEDEnv() ged_env = gedlibpy.GEDEnv()
ged_env.set_edit_cost(options['edit_cost'], edit_cost_constant=options['edit_cost_constants']) ged_env.set_edit_cost(options['edit_cost'], edit_cost_constant=options['edit_cost_constants'])
ged_env.add_nx_graph(g1, '') ged_env.add_nx_graph(g1, '')
@@ -124,6 +125,8 @@ def compute_geds_cml(graphs, options={}, sort=True, parallel=False, verbose=True




def compute_geds(graphs, options={}, sort=True, parallel=False, verbose=True): def compute_geds(graphs, options={}, sort=True, parallel=False, verbose=True):
from gklearn.gedlib import librariesImport, gedlibpy

# initialize ged env. # initialize ged env.
ged_env = gedlibpy.GEDEnv() ged_env = gedlibpy.GEDEnv()
ged_env.set_edit_cost(options['edit_cost'], edit_cost_constant=options['edit_cost_constants']) ged_env.set_edit_cost(options['edit_cost'], edit_cost_constant=options['edit_cost_constants'])


Loading…
Cancel
Save