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
No related branches found
No related tags found
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