Как сериализовать матрицу КСО

Question

У меня есть csr_matrix объекты, рассчитанные по TfidfTransformer.transform.Я хочу сериализовать или снять сериализацию:

tf_csrmtrix = TfidfTransformer.transform(data)
json.dumps(tf_csrmtrix)
TypeError: Object of type 'csr_matrix' is not JSON serializable

Как мне работать с этим королем данных: матрицы csr?

Answer 1

Если двоичный формат сериализации поможет, я предложу модуль pickle

просто import pickle и используйте pickle.dumps там, где вы использовали json.dumps (обратите внимание, что это дает байтовый объект, а не строку, как это делает json)

вы должны быть в состоянии записать в файл, в БД или через какой-либо сокет и загрузить его (используя pickle) позже, чтобы получить точно такой же объект