Added calculation notes for partitions ratios

fhnw-ivy · Jun 6, 2024 · d7b15a2 · d7b15a2
1 parent 84bb577
commit d7b15a2
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/prep_datasets.py b/src/prep_datasets.py
@@ -79,6 +79,9 @@ def parse_fractions_string(fractions_str):
     return [float(frac) for frac in fractions_str.split(",")]
 
 
+# len(dataset) = 3'600'000
+# 1 / 1440 * len(dataset) = 2500
+# 1 / 10 * 1 / 1440 * len(dataset) = 250
 @app.command()
 def prepare_dataset(
         dev_set_fraction: float = typer.Option(1 / 1440,