Fix fake data generation

Quemandoacromo · Jun 5, 2020 · ff3ac62 · ff3ac62
1 parent e22c64a
commit ff3ac62
Showing 1 changed file with 17 additions and 17 deletions.
diff --git a/tests/test_eval.py b/tests/test_eval.py
@@ -26,28 +26,28 @@
 from spleeter.utils.configuration import load_configuration
 
 res_4stems = {  "vocals": {
-                    "SDR": -0.009,
-                    "SAR": -19.044,
-                    "SIR": -4.072,
-                    "ISR": -0.000
+                    "SDR": -0.007,
+                    "SAR": -19.231,
+                    "SIR": -4.528,
+                    "ISR": 0.000
                 },
                 "drums": {
-                    "SDR": -0.066,
-                    "SAR": -14.294,
-                    "SIR": -4.908,
-                    "ISR": 0.002
+                    "SDR": -0.071,
+                    "SAR": -14.496,
+                    "SIR": -4.987,
+                    "ISR": 0.001
                 },
                 "bass":{
-                    "SDR": -0.000,
-                    "SAR": -6.364,
-                    "SIR": -9.396,
+                    "SDR": -0.001,
+                    "SAR": -12.426,
+                    "SIR": -7.198,
                     "ISR": -0.001
                 },
                 "other":{
-                    "SDR": -1.464,
-                    "SAR": -14.893,
-                    "SIR": -4.762,
-                    "ISR": -0.027
+                    "SDR": -1.453,
+                    "SAR": -14.899,
+                    "SIR": -4.678,
+                    "ISR": -0.015
                 }
             }
 
@@ -58,10 +58,10 @@ def generate_fake_eval_dataset(path):
     fs = 44100
     duration = 3
     n_channels = 2
+    rng = np.random.RandomState(seed=0)
     for song in range(n_songs):
         song_path = join(path, "test", f"song{song}")
         makedirs(song_path, exist_ok=True)
-        rng = np.random.RandomState(seed=0)
         for instr in ["mixture", "vocals", "bass", "drums", "other"]:
             filename = join(song_path, f"{instr}.wav")
             data = rng.rand(duration*fs, n_channels)-0.5
@@ -76,4 +76,4 @@ def test_evaluate(path="FAKE_MUSDB_DIR"):
     metrics = evaluate.entrypoint(arguments, params)
     for instrument, metric in metrics.items():
         for metric, value in metric.items():
-            assert np.allclose(np.median(value), res_4stems[instrument][metric], atol=1e-3)
+            assert np.allclose(np.median(value), res_4stems[instrument][metric], atol=1e-3)