Как уменьшить размер типа данных Numpy ⇐ Python

[Anonymous]

Я использую Python для вычисления косинусного подобия.

Код: Выделить всё

similarity_matrix = cosine_similarity(tfidf_matrix)

Проблема в том, что я получаю эту ошибку

Код: Выделить всё

MemoryError: Unable to allocate 44.8 GiB for an array with shape (6011226750,) and data type float64

Я не думаю, что мне нужен float64 для этой операции, поскольку двух цифр после десятичной точки должно быть достаточно.
Есть ли способ изменить тип данных cosine_similarity на меньший?


Подробнее здесь: https://stackoverflow.com/questions/791 ... -data-type