lil_matrix()

看代码的时候遇到的,简单记录记录一下。

这是一种用于逐步构建稀疏矩阵的结构(官方文档中写的),换句话说这是一种存储稀疏矩阵的方式。该稀疏矩阵通过两个list存储:

rows = [list([ ]) list([ ]) list([ ]) ... list([ ])] & datas=[list([ ]) list([ ]) list([ ]) ... list([ ])]

看着很唬人,其实很简单:rows中的每个list存储对应行中非0元素所在的列,value中的每个list代表对应行中非0元素的值。

lil内置了两种方法:

  • lil.rows(): 返回rows
  • lil.data():返回datas

示例:

>>> from scipy import sparse
>>> l = sparse.lil_matrix((6,5))
>>> l[2,3] = 1
>>> l[3,4] = 2
>>> l[3,1] = 3
>>> l[4,4] = 10
>>> print(l.toarray())
[[ 0.  0.  0.  0.  0.]
 [ 0.  0.  0.  0.  0.]
 [ 0.  0.  0.  1.  0.]
 [ 0.  3.  0.  0.  2.]
 [ 0.  0.  0.  0. 10.]
 [ 0.  0.  0.  0.  0.]]

>>> print(l.rows)
[list([]) list([]) list([3]) list([1, 4]) list([4]) list([])]
>>> print(l.data)
[list([]) list([]) list([1.0]) list([3.0, 2.0]) list([10.0]) list([])]

如下是一段构建(user,positive_item)对的代码:

…… 
user_item_matrix = lil_matrix(user_item_matrix)
user_to_positive_set = {u: set(row) for u, row in enumerate(user_item_matrix.rows)} 

你可能感兴趣的:(tensorflow2,算法)