我们可以将tdqm进度栏与t-sne功能一起使用吗？

Question

我正在运行程序，使用t-sne将MNIST数据的维数从784维减小到2维。

程序需要很长时间才能完成日志，我想使用tqdm进度条跟踪进度。

我想知道如何使用t-sne功能添加tqdm进度栏

tqdm在循环中工作正常。

我不知道如何将其用于功能。

# TSNE
from sklearn.manifold import TSNE

# Picking the top 1000 points as TSNE takes a lot of time for 15K points
data_1000 = standardized_data[0:30000,:]
labels_1000 = labels[0:30000]

# configuring the parameteres
# the number of components = 2
# default perplexity = 30
# default learning rate = 200
# default Maximum number of iterations for the optimization = 1000
model = TSNE(n_components=2, random_state=0, perplexity=200,n_iter=5000)

# I want to keep track of progress for function 
tsne_data = model.fit_transform(data_1000)

Answer 1

您必须将进展情况通知tqdm。在您的情况下，TSNE是一个外部函数，所有迭代都在TSNE中完成。除非您去更改TSNE代码以通知tqdm，否则无法在此过程中使用tqdm。