Skip to content

Commit a54f5d4

Browse files
committed
Added TSQR example
1 parent 42909d8 commit a54f5d4

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

dislib/decomposition/tsqr/base.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -62,9 +62,11 @@ def tsqr(a: Array, mode="complete", indexes=None):
6262
>>> from dislib.decomposition import tsqr
6363
>>>
6464
>>> def main():
65-
>>> a = ds.random_array(shape=(125000, 12500), block_size=(2500, 2500))
65+
>>> a = ds.random_array(shape=(125000, 12500),
66+
>>> block_size=(2500, 2500))
6667
>>> q, r = tsqr(a, mode="reduced")
67-
>>> print(np.allclose(np.dot(q.collect(), r.collect()), a.collect()))
68+
>>> print(np.allclose(np.dot(q.collect(),
69+
>>> r.collect()), a.collect()))
6870
>>>
6971
>>> if __name__ == "__main__":
7072
>>> main()

0 commit comments

Comments
 (0)