python - 如何序列化 CSR 矩阵

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），以获得完全相同的对象