如何估计一个cosine_similarity矩阵所需的内存

Question

我试图通过sklearn.metrics.pairwise.cosine_similarity计算cosine_similarity矩阵。输出矩阵可能是60k*50k。然而，它返回一个内存为32G的MemoryError。我正在考虑向服务器添加更多内存，但是我想估计一下我真正需要多少内存。

谢谢。

Answer 1

所需内存的较好估计为60,000 * 50,000 * 4字节= 12 gb。