Блог

Я запускаю программу для уменьшения размера данных MNIST с 784-dim до 2-dim с использованием t-sne.

Завершение программы занимает очень много времени, и я хочу отслеживать прогресс с помощью индикатора выполнения tqdm.

Я хочу знать, как добавить индикатор выполнения tqdm с помощью функции t-sne. 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)
  

Вы должны уведомлять tqdm о ходе выполнения. В вашем случае TSNE это внешняя функция, в которой выполняется вся итерация TSNE . Если вы не измените TSNE код на notify tqdm , нет никакого способа использовать tqdm этот процесс.