python - 记录 T-SNE 训练？

Question

我们可以记录库的训练如gensim使用

import logging
logger = logging.getLogger()
logger.setLevel(logging.INFO)

这在运行 T-SNE 时不起作用(来自 sklearn.manifold )。由于这可能需要很长时间，因此记录某种形式会很好。我已经浏览了文档，但似乎无法找到另一种方法来记录此培训。有没有办法记录这个，或者不可能？

Answer 1

那是因为 sklearn code没有任何日志记录。 contributing指南对此没有任何标准方法。

但是，您可以使用详细标志来获取有关该类正在执行的操作的更多信息。

如果您像这样使用标志:

import numpy as np
from sklearn.manifold import TSNE

X = np.array([[0, 0, 0], [0, 1, 1], [1, 0, 1], [1, 1, 1]])

tsne = TSNE(n_components=2, verbose=1) # note the verbose flag

X_embedded = tsne.fit_transform(X)
X_embedded.shape(4, 2)

输出将是

[t-SNE] Computing 3 nearest neighbors...
[t-SNE] Indexed 4 samples in 0.000s...
[t-SNE] Computed neighbors for 4 samples in 0.000s...
[t-SNE] Computed conditional probabilities for sample 4 / 4
[t-SNE] Mean sigma: 1125899906842624.000000
[t-SNE] KL divergence after 250 iterations with early exaggeration: 31.763723
[t-SNE] KL divergence after 650 iterations: 0.048727

该代码改编自 sklearn TSNE code example .

请注意，输出是标准输出，而不是类似的文件。来自详细标志的所有消息都是使用打印功能写入的。