__author__ = 'aymgal'
from coolest.template.classes.regularization import Regularization
from coolest.template.classes.parameter import HyperParameter
from coolest.template.classes.parameter import DefinitionRange
__all__ = [
'PixelEntropy',
'PixelRidge',
'PixelLasso',
'PixelGradient',
'PixelCurvature',
'PixelExponentialKernel',
'PixelGaussianKernel',
'PixelStarlet',
'PixelBLWavelet',
'PixelPositivity',
'DefaultForPixelated',
]
SUPPORTED_CHOICES = __all__
[docs]
class PixelEntropy(Regularization):
def __init__(self, **kwargs):
description = "Entropy regularization"
parameters = {
'lambda': HyperParameter("Lagrange parameter for entropy regularization",
DefinitionRange(min_value=0.0),
latex_str=r"$\lambda_{\rm e}$"),
}
super().__init__(description, parameters, **kwargs)
[docs]
class PixelRidge(Regularization):
def __init__(self, **kwargs):
description = "Ridge regularization (L2 norm)"
parameters = {
'lambda': HyperParameter("Lagrange parameter for Ridge norm regularization",
DefinitionRange(min_value=0.0),
latex_str=r"$\lambda_{\rm ridge}$"),
}
super().__init__(description, parameters, **kwargs)
[docs]
class PixelLasso(Regularization):
def __init__(self, **kwargs):
description = "Lasso regularization (L1 norm)"
parameters = {
'lambda': HyperParameter("Lagrange parameter for Lasso norm regularization",
DefinitionRange(min_value=0.0),
latex_str=r"$\lambda_{\rm lasso}$"),
}
super().__init__(description, parameters, **kwargs)
[docs]
class PixelGradient(Regularization):
def __init__(self, **kwargs):
description = "Gradient regularization (first spatial derivative)"
parameters = {
'lambda': HyperParameter("Lagrange parameter for gradient regularization",
DefinitionRange(min_value=0.0),
latex_str=r"$\lambda_{\rm grad}$"),
}
super().__init__(description, parameters, **kwargs)
[docs]
class PixelCurvature(Regularization):
def __init__(self, **kwargs):
description = "Curvature regularization (second spatial derivative)"
parameters = {
'lambda': HyperParameter("Lagrange parameter for curvature regularization",
DefinitionRange(min_value=0.0),
latex_str=r"$\lambda_{\rm curv}$"),
}
super().__init__(description, parameters, **kwargs)
[docs]
class PixelExponentialKernel(Regularization):
def __init__(self, **kwargs):
description = "Exponential kernel regularization"
parameters = {
'lambda': HyperParameter("Lagrange parameter for exponential kernel regularization",
DefinitionRange(min_value=0.0),
latex_str=r"$\lambda_{\rm exp}$"),
}
super().__init__(description, parameters, **kwargs)
[docs]
class PixelGaussianKernel(Regularization):
def __init__(self, **kwargs):
description = "Gaussian kernel regularization"
parameters = {
'lambda': HyperParameter("Lagrange parameter for gaussian kernel regularization",
DefinitionRange(min_value=0.0),
latex_str=r"$\lambda_{\rm gauss}$"),
}
super().__init__(description, parameters, **kwargs)
[docs]
class PixelStarlet(Regularization):
def __init__(self, **kwargs):
description = "Starlets regularization"
parameters = {
'lambda_hf': HyperParameter("Lagrange parameter for first starlet scale (high frequencies)",
DefinitionRange(min_value=0.0),
latex_str=r"$\lambda_{\rm s, hf}$"),
'lambda': HyperParameter("Lagrange parameter from second to last starlet scales (excluding coarse scale)",
DefinitionRange(min_value=0.0),
latex_str=r"$\lambda_{\rm s}$"),
}
super().__init__(description, parameters, **kwargs)
[docs]
class PixelBLWavelet(Regularization):
def __init__(self, **kwargs):
description = "Battle-Lemarie wavelet regularization"
parameters = {
'lambda_hf': HyperParameter("Lagrange parameter for first Battle-Lemarie scale (high frequencies)",
DefinitionRange(min_value=0.0),
latex_str=r"$\lambda_{\rm BL, hf}$"),
}
super().__init__(description, parameters, **kwargs)
[docs]
class PixelPositivity(Regularization):
def __init__(self, **kwargs):
description = "Non-negativity constraint on pixel values"
parameters = {
'lambda': HyperParameter("Lagrange parameter",
DefinitionRange(min_value=0.0),
latex_str=r"$\lambda_{\rm pos}$"),
}
super().__init__(description, parameters, **kwargs)
[docs]
class DefaultForPixelated(Regularization):
def __init__(self, **kwargs):
description = "Default regularization strategy, set by the specific modelling code"
parameters = {
'lambda': HyperParameter("Lagrange parameter",
DefinitionRange(min_value=0.0),
latex_str=r"$\lambda$"),
}
super().__init__(description, parameters, **kwargs)
