run.py

from src.run_modules import *
import argparse
import inspect


def run(stations=['DEBW107', 'DEBY081', 'DEBW013', 'DEBW076', 'DEBW087', 'DEBW001'],
        station_type='background',
        trainable=False, create_new_model=True,
        window_history_size=6,
        experiment_date="testrun",
        network=None,
        variables=None, statistics_per_var=None,
        start=None, end=None,
        target_var="o3", target_dim=None,
        window_lead_time=None,
        dimensions=None,
        interpolate_method=None, interpolate_dim=None, limit_nan_fill=None,
        train_start=None, train_end=None, val_start=None, val_end=None, test_start=None, test_end=None,
        use_all_stations_on_all_data_sets=True, fraction_of_train=None,
        experiment_path=None, plot_path=None, forecast_path=None, bootstrap_path=None, overwrite_local_data=None,
        sampling="daily",
        permute_data_on_training=False, extreme_values=None, extremes_on_right_tail_only=None,
        transformation=None,
        train_min_length=None, val_min_length=None, test_min_length=None,
        evaluate_bootstraps=True, number_of_bootstraps=None, create_new_bootstraps=False,
        plot_list=None,
        model=None,
        batch_size=None,
        epochs=None):

    params = inspect.getfullargspec(ExperimentSetup).args
    kwargs = {k: v for k, v in locals().items() if k in params}

    parser = argparse.ArgumentParser()
    parser.add_argument('--experiment_date', metavar='--exp_date', type=str, default="testrun",
                        help="set experiment date as string")
    args = parser.parse_args()

    with RunEnvironment():
        ExperimentSetup(args, **kwargs)
        PreProcessing()
        PartitionCheck()
        ModelSetup()
        Training()
        PostProcessing()


if __name__ == "__main__":

    run()