コードを見ていて見つけた場合は、記録してください。
これは、疎行列を徐々に構築するための構造 (公式ドキュメントに記載されています)、言い換えれば、疎行列を格納する方法です。スパース行列は 2 つのリストによって保存されます。
rows = [ list([ ]) list([ ]) list([ ]) ... list([ ]) ] & datas= [ list([ ]) list([ ]) list([ ]) ... list ([ ]) ]
非常にハッタリのように見えますが、実際は非常に単純です。行の各リストには、対応する行の非ゼロ要素が配置されている列が格納され、value の各リストは、対応する行の非ゼロ要素の値を表します。 。
lil には 2 つのメソッドが組み込まれています。
- lil.rows(): 行を返します
- lil.data(): データを返します
例:
>>> 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)}