From 368483bda22d9da6af7e650a632f20e7b4b5bc16 Mon Sep 17 00:00:00 2001 From: Seanghay Yath Date: Mon, 8 Jan 2024 17:53:51 +0700 Subject: [PATCH] Update visualize.py --- visualize.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/visualize.py b/visualize.py index 442d589..409200b 100755 --- a/visualize.py +++ b/visualize.py @@ -6,6 +6,7 @@ from pathlib import Path from warnings import filterwarnings +import numpy as np import matplotlib.pyplot as plt import seaborn as sns import torch @@ -52,7 +53,7 @@ def visualize(data_dirs, wav2mel_path, checkpoint_path, output_path): emb = emb.detach().cpu().numpy() embs.append(emb) - embs = np.array(emb) + embs = np.array(embs) tsne = TSNE(n_components=2, verbose=1, perplexity=40, n_iter=300) transformed = tsne.fit_transform(embs)