From d529e060524b7dad42e0682eacc5fafda3617dcc Mon Sep 17 00:00:00 2001 From: jajupmochi Date: Fri, 16 Oct 2020 18:34:33 +0200 Subject: [PATCH] Update MPG. --- gklearn/preimage/median_preimage_generator.py | 2 +- gklearn/tests/test_median_preimage_generator.py | 6 +++++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/gklearn/preimage/median_preimage_generator.py b/gklearn/preimage/median_preimage_generator.py index dae2b0a..7cb362f 100644 --- a/gklearn/preimage/median_preimage_generator.py +++ b/gklearn/preimage/median_preimage_generator.py @@ -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 diff --git a/gklearn/tests/test_median_preimage_generator.py b/gklearn/tests/test_median_preimage_generator.py index a376c7a..b8b294e 100644 --- a/gklearn/tests/test_median_preimage_generator.py +++ b/gklearn/tests/test_median_preimage_generator.py @@ -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 \ No newline at end of file + assert False, exception + + +if __name__ == '__main__': + test_median_preimage_generator() \ No newline at end of file