LSST Applications  21.0.0-131-g8cabc107+528f53ee53,22.0.0+00495a2688,22.0.0+0ef2527977,22.0.0+11a2aa21cd,22.0.0+269b7e55e3,22.0.0+2c6b6677a3,22.0.0+64c1bc5aa5,22.0.0+7b3a3f865e,22.0.0+e1b6d2281c,22.0.0+ff3c34362c,22.0.1-1-g1b65d06+c95cbdf3df,22.0.1-1-g7058be7+1cf78af69b,22.0.1-1-g7dab645+2a65e40b06,22.0.1-1-g8760c09+64c1bc5aa5,22.0.1-1-g949febb+64c1bc5aa5,22.0.1-1-ga324b9c+269b7e55e3,22.0.1-1-gf9d8b05+ff3c34362c,22.0.1-10-g781e53d+9b51d1cd24,22.0.1-10-gba590ab+b9624b875d,22.0.1-13-g76f9b8d+2c6b6677a3,22.0.1-14-g22236948+57af756299,22.0.1-18-g3db9cf4b+9b7092c56c,22.0.1-18-gb17765a+2264247a6b,22.0.1-2-g8ef0a89+2c6b6677a3,22.0.1-2-gcb770ba+c99495d3c6,22.0.1-24-g2e899d296+4206820b0d,22.0.1-3-g7aa11f2+2c6b6677a3,22.0.1-3-g8c1d971+f253ffa91f,22.0.1-3-g997b569+ff3b2f8649,22.0.1-4-g1930a60+6871d0c7f6,22.0.1-4-g5b7b756+6b209d634c,22.0.1-6-ga02864e+6871d0c7f6,22.0.1-7-g3402376+a1a2182ac4,22.0.1-7-g65f59fa+54b92689ce,master-gcc5351303a+e1b6d2281c,w.2021.32
LSST Data Management Base Package
Public Member Functions | List of all members
lsst.cp.pipe.ptc.astierCovFitParameters.FitParameters Class Reference
Inheritance diagram for lsst.cp.pipe.ptc.astierCovFitParameters.FitParameters:

Public Member Functions

def __init__ (self, groups)
 
def copy (self)
 
def fix (self, keys=slice(None), val=None)
 
def release (self, keys=slice(None))
 
def indexof (self, indices=slice(None))
 
def __getitem__ (self, args)
 
def __setitem__ (self, args, val)
 
def free (self)
 
def free (self, val)
 
def full (self)
 
def full (self, val)
 
def __repr__ (self)
 

Detailed Description

 Manages a vector of fit parameters with the possibility to mark a subset of
them as fixed to a given value.

The parameters can be organized in named slices (block of contiguous
values) accessible through indexing by their name as in `StructArray`.

>>> p_salt = FitParameters(['X0', 'X1', 'Color', 'Redshift'])
>>> p_dice = FitParameters([('alpha', 2), ('S', 10), ('dSdT', 10), 'idark'])

It is possible to modify the parameters in place. Using the indexing
of slices by name simplifies somewhat the operations, as one does
not need to care about the position of a slice within the entire
parameter vector:

>>> p_dice['idark'][0] = -1.0232E-12
>>> p_dice['S'][4] = 25.242343E-9
>>> p_dice['dSdT'][:] = 42.
>>> print p_dice
alpha: array([ 0.,  0.])
S: array([  0.00000000e+00,   0.00000000e+00,   0.00000000e+00,
         0.00000000e+00,   2.52423430e-08,   0.00000000e+00,
         0.00000000e+00,   0.00000000e+00,   0.00000000e+00,
         0.00000000e+00])
dSdT: array([ 42.,  42.,  42.,  42.,  42.,  42.,  42.,  42.,  42.,  42.])
idark: array([ -1.02320000e-12])

It is also possible to mark parameters as fixed to a value.
>>> p_dice.fix(0, 12.)

Value is optional. The above is equivalent to:
>>> p_dice[0] = 12.
>>> p_dice.fix(0)

Again named slices simplifies the operations:
>>> p_dice['S'].fix([0, -1], 12.)
>>> p_dice['dSdT'].fix([0, -1])

One can fix entire slices at once:
>>> p_dice['idark'].fix()
>>> p_salt['Redshift'].fix(val=0.23432)

The property ``full'' give access to the vector of parameters. The
property "free" gives access to the free parameters:
>>> print len(p_dice.free), len(p_dice.full)
17 23

Note that free relies on fancy indexing. Access thus trigger a
copy. As a consequence, the following will not actually alter the
data:
>>> p_dice.free[0] = 12.
>>> print p_dice.free[0]
0.0

It is still possible to set slices of free parameters as a
contiguous vector. For example:
>>> p_dice['S'].free = 12.
>>> print p_dice['S'].free
[ 12.  12.  12.  12.  12.  12.  12.  12.]

>>> p_dice[:5].free = 4.
>>> print p_dice[:5].free
[ 4.  4.  4.]

In particular, the typical use case which consists in updating the
free parameters with the results of a fit works as expected:
>>> p = np.arange(len(p_dice.free))
>>> p_dice.free = p

Last the class provide a convenience function that return the index
of a subset of parameters in the global free parameters vector, and
-1 for fixed parameters:
>>> print p_dice['dSdT'].indexof()
[-1  9 10 11 12 13 14 15 16 -1]
>>> print p_dice['dSdT'].indexof([1,2])
[ 9 10]

Definition at line 164 of file astierCovFitParameters.py.

Constructor & Destructor Documentation

◆ __init__()

def lsst.cp.pipe.ptc.astierCovFitParameters.FitParameters.__init__ (   self,
  groups 
)

Definition at line 241 of file astierCovFitParameters.py.

241  def __init__(self, groups):
242  struct = Structure(groups)
243  self._free = StructArray(np.ones(len(struct), dtype='bool'), struct)
244  self._pars = StructArray(np.zeros(len(struct), dtype='float'), struct)
245  self._index = StructArray(np.zeros(len(struct), dtype='int'), struct)
246  self._struct = struct
247  self._reindex()
248  self._base = self
249 

Member Function Documentation

◆ __getitem__()

def lsst.cp.pipe.ptc.astierCovFitParameters.FitParameters.__getitem__ (   self,
  args 
)

Definition at line 276 of file astierCovFitParameters.py.

276  def __getitem__(self, args):
277  # Prevent conversion to scalars
278  if isinstance(args, int):
279  args = slice(args, args + 1)
280  new = FitParameters.__new__(FitParameters)
281  new._free = self._free[args]
282  new._pars = self._pars[args]
283  new._index = self._index[args]
284  new._base = self._base
285  return new
286 

◆ __repr__()

def lsst.cp.pipe.ptc.astierCovFitParameters.FitParameters.__repr__ (   self)

Definition at line 306 of file astierCovFitParameters.py.

306  def __repr__(self):
307  if hasattr(self, '_struct'):
308  s = "\n".join(['%s: %s' % (key, repr(self._pars[key])) for key in self._struct])
309  else:
310  s = repr(self._pars)
311  return s

◆ __setitem__()

def lsst.cp.pipe.ptc.astierCovFitParameters.FitParameters.__setitem__ (   self,
  args,
  val 
)

Definition at line 287 of file astierCovFitParameters.py.

287  def __setitem__(self, args, val):
288  self._pars[args] = val
289 

◆ copy()

def lsst.cp.pipe.ptc.astierCovFitParameters.FitParameters.copy (   self)

Definition at line 250 of file astierCovFitParameters.py.

250  def copy(self):
251  cop = FitParameters(self._struct)
252  cop._free = StructArray(np.copy(self._free), cop._struct)
253  cop._pars = StructArray(np.copy(self._pars), cop._struct)
254  cop._index = StructArray(np.copy(self._index), cop._struct)
255  cop._reindex()
256  cop._base = cop
257  return cop
258 

◆ fix()

def lsst.cp.pipe.ptc.astierCovFitParameters.FitParameters.fix (   self,
  keys = slice(None),
  val = None 
)

Definition at line 263 of file astierCovFitParameters.py.

263  def fix(self, keys=slice(None), val=None):
264  self._free[keys] = False
265  if val is not None:
266  self._pars[keys] = val
267  self._base._reindex()
268 

◆ free() [1/2]

def lsst.cp.pipe.ptc.astierCovFitParameters.FitParameters.free (   self)

Definition at line 291 of file astierCovFitParameters.py.

291  def free(self):
292  return self._pars[self._free]
293 

◆ free() [2/2]

def lsst.cp.pipe.ptc.astierCovFitParameters.FitParameters.free (   self,
  val 
)

Definition at line 295 of file astierCovFitParameters.py.

295  def free(self, val):
296  self._pars[self._free] = val
297 

◆ full() [1/2]

def lsst.cp.pipe.ptc.astierCovFitParameters.FitParameters.full (   self)

Definition at line 299 of file astierCovFitParameters.py.

299  def full(self):
300  return self._pars
301 

◆ full() [2/2]

def lsst.cp.pipe.ptc.astierCovFitParameters.FitParameters.full (   self,
  val 
)

Definition at line 303 of file astierCovFitParameters.py.

303  def full(self, val):
304  self._pars = val
305 

◆ indexof()

def lsst.cp.pipe.ptc.astierCovFitParameters.FitParameters.indexof (   self,
  indices = slice(None) 
)

Definition at line 273 of file astierCovFitParameters.py.

273  def indexof(self, indices=slice(None)):
274  return self._index[indices]
275 

◆ release()

def lsst.cp.pipe.ptc.astierCovFitParameters.FitParameters.release (   self,
  keys = slice(None) 
)

Definition at line 269 of file astierCovFitParameters.py.

269  def release(self, keys=slice(None)):
270  self._free[keys] = True
271  self._base._reindex()
272 

The documentation for this class was generated from the following file: