Skip to content
Snippets Groups Projects
Commit 587f606d authored by lukas leufen's avatar lukas leufen
Browse files

prepare experiment to use bootstraps

parent ebc4ff73
Branches
Tags
3 merge requests!59Develop,!52implemented bootstraps,!34Pair issue048 feat create shuffled data
Pipeline #29190 passed with warnings
...@@ -145,6 +145,13 @@ def set_experiment_name(experiment_date=None, experiment_path=None): ...@@ -145,6 +145,13 @@ def set_experiment_name(experiment_date=None, experiment_path=None):
return experiment_name, experiment_path return experiment_name, experiment_path
def set_bootstrap_path(bootstrap_path, data_path, sampling):
if bootstrap_path is None:
bootstrap_path = os.path.join(data_path, "..", f"bootstrap_{sampling}")
check_path_and_create(bootstrap_path)
return bootstrap_path
class PyTestRegex: class PyTestRegex:
"""Assert that a given string meets some expectations.""" """Assert that a given string meets some expectations."""
......
...@@ -33,13 +33,17 @@ class ExperimentSetup(RunEnvironment): ...@@ -33,13 +33,17 @@ class ExperimentSetup(RunEnvironment):
window_lead_time=None, dimensions=None, interpolate_dim=None, interpolate_method=None, window_lead_time=None, dimensions=None, interpolate_dim=None, interpolate_method=None,
limit_nan_fill=None, train_start=None, train_end=None, val_start=None, val_end=None, test_start=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, trainable=False, fraction_of_train=None, test_end=None, use_all_stations_on_all_data_sets=True, trainable=False, fraction_of_train=None,
experiment_path=None, plot_path=None, forecast_path=None, overwrite_local_data=None): experiment_path=None, plot_path=None, forecast_path=None, overwrite_local_data=None, sampling="daily",
bootstrap_path=None):
# create run framework # create run framework
super().__init__() super().__init__()
# experiment setup # experiment setup
self._set_param("data_path", helpers.prepare_host()) self._set_param("data_path", helpers.prepare_host())
data_path = self.data_store.get("data_path", "general")
bootstrap_path = helpers.set_bootstrap_path(bootstrap_path, data_path, sampling)
self._set_param("bootstrap_path", bootstrap_path)
self._set_param("trainable", trainable, default=False) self._set_param("trainable", trainable, default=False)
self._set_param("fraction_of_training", fraction_of_train, default=0.8) self._set_param("fraction_of_training", fraction_of_train, default=0.8)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment