diff --git a/run_bnn.py b/run_bnn.py index 7667b027a4293a611a45355be635435e3fb0f3b1..f1ed70ab3719ec00ce975b199d7b40798261c9cb 100644 --- a/run_bnn.py +++ b/run_bnn.py @@ -33,6 +33,16 @@ def main(parser_args): stats_per_var = {'o3': 'dma8eu', 'relhum': 'average_values', 'temp': 'maximum', 'u': 'average_values', 'v': 'average_values', 'no': 'dma8eu', 'no2': 'dma8eu', 'cloudcover': 'average_values', 'pblheight': 'maximum'} + transformation = {'o3': {'method': 'standardise'}, + 'relhum': {'method': 'min_max'}, + 'temp': {'method': 'standardise'}, + 'u': {'method': 'standardise'}, + 'v': {'method': 'standardise'}, + 'no': {'method': 'standardise'}, + 'no2': {'method': 'standardise'}, + 'cloudcover': {'method': 'min_max'}, + 'pblheight': {'method': 'standardise'} + } workflow = DefaultWorkflow( # stations=load_stations(), #stations=["DEBW087","DEBW013", "DEBW107", "DEBW076"], stations=load_stations(2), @@ -41,6 +51,7 @@ def main(parser_args): window_history_size=6, epochs=100, batch_size=1024, + transformation=transformation, train_model=False, create_new_model=True, network="UBA", evaluate_feature_importance=False, # plot_list=["PlotCompetitiveSkillScore"], # competitors=["test_model", "test_model2"],