如何序列化 CSR 矩阵

【问题标题】：How to Serialize CSR Matrix如何序列化 CSR 矩阵
【发布时间】：2019-11-10 10:26:06
【问题描述】：

我有一个从TfidfTransformer.transform 计算的csr_matrix 对象。我想序列化或反序列化：

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

如何处理这个数据之王：csr 矩阵？

【问题讨论】：

【解决方案1】：

如果二进制序列化格式可以解决问题，我建议pickle module

只需 import pickle 并在您使用 json.dumps 的地方使用 pickle.dumps（请注意，这给出了一个字节对象，而不是像 json 那样的字符串）

您应该能够写入文件、数据库或通过某些套接字，并稍后加载它（使用 pickle），以获得完全相同的对象

【讨论】：