From dde77c42e3de8e17093361e7b14b0a58d556cc13 Mon Sep 17 00:00:00 2001 From: "v.gramlich1" <v.gramlichfz-juelich.de> Date: Fri, 23 Jul 2021 08:24:45 +0200 Subject: [PATCH] Coded the structure for PlotOversamplingContingency --- mlair/run_modules/post_processing.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/mlair/run_modules/post_processing.py b/mlair/run_modules/post_processing.py index 6acfcfbb..c5f5b2d3 100644 --- a/mlair/run_modules/post_processing.py +++ b/mlair/run_modules/post_processing.py @@ -309,13 +309,11 @@ class PostProcessing(RunEnvironment): try: if (self.data_store.get('oversampling_method')=='bin_oversampling') and ( "PlotOversamplingContingency" in plot_list): - bin_edges = self.data_store.get('oversampling_bin_edges') - oversampling_rates = self.data_store.get('oversampling_rates_capped','train') predictions = None labels = None - PlotOversampling(predictions, labels, plot_folder=self.plot_path) + PlotOversamplingContingency(predictions, labels, plot_folder=self.plot_path) except Exception as e: - logging.error(f"Could not create plot OversamplingPlots due to the following error: {e}") + logging.error(f"Could not create plot OversamplingContingencyPlots due to the following error: {e}") try: if (self.data_store.get('oversampling_method')=='bin_oversampling') and ( -- GitLab