From 4b2fdc13a43e29b2bab4bc37ee44e50a37c6d05b Mon Sep 17 00:00:00 2001 From: narugo1992 Date: Mon, 2 Sep 2024 00:42:35 +0800 Subject: [PATCH] dev(narugo): xxx --- zoo/pyannote/embedding/encode.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/zoo/pyannote/embedding/encode.py b/zoo/pyannote/embedding/encode.py index 3fc52ee..c4bf3d7 100644 --- 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)