dev(narugo): xxx

deepghs · Sep 1, 2024 · 4b2fdc1 · 4b2fdc1
1 parent aaf08d6
commit 4b2fdc1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/zoo/pyannote/embedding/encode.py b/zoo/pyannote/embedding/encode.py
@@ -15,8 +15,8 @@
 if __name__ == '__main__':
     files = []
     names = []
-    dataset_dir = get_testfile('assets', 'speakers')
-    # dataset_dir = '/data/arknights_jp_nested'
+    # dataset_dir = get_testfile('assets', 'speakers')
+    dataset_dir = '/data/arknights_jp_nested'
     for name in os.listdir(dataset_dir):
         new_files = glob.glob(os.path.join(dataset_dir, name, '*.wav'))
         files.extend(new_files)
@@ -77,7 +77,7 @@ def find_optimal_threshold(positive_scores, negative_scores, max_samples: int =
 
         # 计算每个可能的阈值的F1分数
         scores, f1s, ps, rs = [], [], [], []
-        for i, threshold in enumerate(scs):
+        for i, threshold in enumerate(tqdm(scs)):
             tp -= np.sum(pos_ranks == i)
             fp -= np.sum(neg_ranks == i)