Merge pull request #4 from JonathanCrabbe/drought

US-Droughts Dataset
JonathanCrabbe · Jan 15, 2024 · bdb2970 · bdb2970
2 parents dcf0a1e + 6898e56
commit bdb2970
Show file tree

Hide file tree

Showing 6 changed files with 7,334 additions and 3 deletions.
diff --git a/cmd/conf/datamodule/usdroughts.yaml b/cmd/conf/datamodule/usdroughts.yaml
@@ -0,0 +1,6 @@
+_target_: fdiff.dataloaders.datamodules.USDroughtsDatamodule
+data_dir: ${hydra:runtime.cwd}/data
+random_seed: ${random_seed}
+fourier_transform: ${fourier_transform}
+standardize: ${standardize}
+batch_size: 64
diff --git a/cmd/conf/metrics/default.yaml b/cmd/conf/metrics/default.yaml
@@ -1,6 +1,7 @@
 _target_: fdiff.sampling.metrics.MetricCollection
 _partial_: true
 include_baselines: true
+include_spectral_density: true
 metrics:
   - _target_: fdiff.sampling.metrics.SlicedWasserstein
     _partial_: true