Browse Source

Update MPG.

v0.2.x
jajupmochi 4 years ago
parent
commit
d529e06052
2 changed files with 6 additions and 2 deletions
  1. +1
    -1
      gklearn/preimage/median_preimage_generator.py
  2. +5
    -1
      gklearn/tests/test_median_preimage_generator.py

+ 1
- 1
gklearn/preimage/median_preimage_generator.py View File

@@ -183,7 +183,7 @@ class MedianPreimageGenerator(PreimageGenerator):
results['mge']['num_decrease_order'] = self._mge.get_num_times_order_decreased()
results['mge']['num_increase_order'] = self._mge.get_num_times_order_increased()
results['mge']['num_converged_descents'] = self._mge.get_num_converged_descents()
results['ged_matrix_set_median'] = self._mge.ged_matrix_set_median_tmp
# results['ged_matrix_set_median'] = self._mge.ged_matrix_set_median_tmp
return results



+ 5
- 1
gklearn/tests/test_median_preimage_generator.py View File

@@ -71,4 +71,8 @@ def test_median_preimage_generator():
try:
generate_median_preimages_by_class(ds_name, mpg_options, kernel_options, ged_options, mge_options, save_results=save_results, save_medians=True, plot_medians=True, load_gm='auto', dir_save=dir_save, irrelevant_labels=irrelevant_labels, edge_required=edge_required, cut_range=range(0, 4))
except Exception as exception:
assert False, exception
assert False, exception
if __name__ == '__main__':
test_median_preimage_generator()

Loading…
Cancel
Save