From 0d218501aca82a10d32d2ee8e51897bc72bd91a9 Mon Sep 17 00:00:00 2001 From: Mathias Date: Mon, 14 Sep 2020 15:53:22 +0200 Subject: [PATCH] Add an option to specify which file is the list of ids --- scripts/evaluations/clustering.py | 18 +++++++++++++++++- 1 file changed, 17 insertions(+), 1 deletion(-) diff --git a/scripts/evaluations/clustering.py b/scripts/evaluations/clustering.py index 60f6286..148bf7f 100644 --- a/scripts/evaluations/clustering.py +++ b/scripts/evaluations/clustering.py @@ -23,6 +23,14 @@ if __name__ == "__main__": parser.add_argument("utt2", type=str, help="file with [utt] [value]") + parser.add_argument("--idsfrom", + type=str, + default="utt2", + choices=[ + "features", + "utt2" + ], + help="from features or from utt2?") parser.add_argument("--prefix", type=str, help="prefix of saved files") @@ -41,7 +49,15 @@ if __name__ == "__main__": features = read_features(args.features) utt2 = read_lst(args.utt2) - ids = list(features.keys()) + # Take id list + if args.idsfrom == "features": + ids = list(features.keys()) + elif args.idsfrom == "utt2": + ids = list(utt2.keys()) + else: + print(f"idsfrom is not good: {args.idsfrom}") + exit(1) + feats = np.vstack([ features[id_] for id_ in ids ]) classes = [ utt2[id_] for id_ in ids ] -- 1.8.2.3