def test_01(self):
lmbda = 3
try:
b = tvl2.TVL2Denoise(self.D, lmbda)
b.solve()
except Exception as e:
print(e)
assert 0
def test_01(self):
lmbda = 1e-1
opt = tvl2.TVL2Denoise.Options(
{'Verbose': False, 'gEvalY': False, 'MaxMainIter': 250,
'rho': 10 * lmbda})
b = tvl2.TVL2Denoise(self.D, lmbda, opt, axes=(0, 1, 2))
X = b.solve()
assert cp.abs(b.itstat[-1].ObjFun - 366.04267554965134) < 1e-3
assert sm.mse(self.U, X) < 1e-3
def test_01(self):
lmbda = 1e-1
opt = tvl2.TVL2Denoise.Options(
{'Verbose': False, 'gEvalY': False, 'MaxMainIter': 300,
'rho': 75 * lmbda})
b = tvl2.TVL2Denoise(self.D, lmbda, opt)
X = b.solve()
assert cp.abs(b.itstat[-1].ObjFun - 32.875710674129564) < 1e-3
assert sm.mse(self.U, X) < 1e-3
def test_05(self):
lmbda = 3
dt = cp.float64
opt = tvl2.TVL2Denoise.Options(
{'Verbose': False, 'MaxMainIter': 20, 'AutoRho':
{'Enabled': True}, 'DataType': dt})
b = tvl2.TVL2Denoise(self.D, lmbda, opt=opt)
b.solve()
assert b.X.dtype == dt
assert b.Y.dtype == dt
assert b.U.dtype == dt
