File tree Expand file tree Collapse file tree 1 file changed +7
-2
lines changed
Expand file tree Collapse file tree 1 file changed +7
-2
lines changed Original file line number Diff line number Diff line change 77import numpy as np
88from numpy .lib .stride_tricks import sliding_window_view
99
10+ # Named tuples used functions.
11+ # compute_svd uses "SVD",
12+ # compute_tlsq uses "TLSQ".
13+ SVD = namedtuple ("SVD" , ["U" , "s" , "V" ])
14+ TLSQ = namedtuple ("TLSQ" , ["X_denoised" , "Y_denoised" ])
15+
1016
1117def _svht (sigma_svd : np .ndarray , rows : int , cols : int ) -> int :
1218 """
@@ -140,7 +146,7 @@ def compute_tlsq(
140146 V = np .linalg .svd (np .append (X , Y , axis = 0 ), full_matrices = False )[- 1 ]
141147 rank = min (tlsq_rank , V .shape [0 ])
142148 VV = V [:rank , :].conj ().T .dot (V [:rank , :])
143- TLSQ = namedtuple ( "TLSQ" , [ "X_denoised" , "Y_denoised" ])
149+
144150 return TLSQ (X .dot (VV ), Y .dot (VV ))
145151
146152
@@ -179,7 +185,6 @@ def compute_svd(
179185 U = U [:, :rank ]
180186 V = V [:, :rank ]
181187 s = s [:rank ]
182- SVD = namedtuple ("SVD" , ["U" , "s" , "V" ])
183188
184189 return SVD (U , s , V )
185190
You can’t perform that action at this time.
0 commit comments