diff --git a/mlair/run_modules/post_processing.py b/mlair/run_modules/post_processing.py index 3dc91cbd54094f116f0d959fb9c845751e998464..21fbb3edfa2c9665db12f7ea6bb0f9310ca324a1 100644 --- a/mlair/run_modules/post_processing.py +++ b/mlair/run_modules/post_processing.py @@ -19,8 +19,8 @@ from mlair.helpers import TimeTracking, statistics, extract_value from mlair.model_modules.linear_model import OrdinaryLeastSquaredModel from mlair.model_modules.model_class import AbstractModelClass from mlair.plotting.postprocessing_plotting import PlotMonthlySummary, PlotStationMap, PlotClimatologicalSkillScore, \ - PlotCompetitiveSkillScore, PlotTimeSeries, PlotBootstrapSkillScore, PlotAvailability, PlotConditionalQuantiles, \ - PlotSeparationOfScales + PlotCompetitiveSkillScore, PlotTimeSeries, PlotBootstrapSkillScore, PlotAvailability, PlotAvailabilityHistogram, \ + PlotConditionalQuantiles, PlotSeparationOfScales from mlair.run_modules.run_environment import RunEnvironment @@ -293,6 +293,10 @@ class PostProcessing(RunEnvironment): if "PlotAvailability" in plot_list: avail_data = {"train": self.train_data, "val": self.val_data, "test": self.test_data} PlotAvailability(avail_data, plot_folder=self.plot_path, time_dimension=time_dimension) + if "PlotAvailabilityHistogram" in plot_list: + avail_data = {"train": self.train_data, "val": self.val_data, "test": self.test_data} + PlotAvailabilityHistogram(avail_data, plot_folder=self.plot_path, )# time_dimension=time_dimension) + def calculate_test_score(self): """Evaluate test score of model and save locally."""