lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure Class Reference

Public Member Functions
def	__init__ (self, parent, xDim, yDim, n=0)
def	xlim (self)
def	ylim (self)
def	zlim (self)
def	guessExtent (self)
def	setExtent (self, x0=None, x1=None, y0=None, y1=None, z0=None, z1=None, lock=True)
def	plotTrack (self)
def	plotRejected (self)
def	plotSurfaces (self)
def	move (self, n)

Public Attributes
	parent
	xDim
	yDim
	j
	i
	yKey
	xKey
	zKey
	slice2d
	sliceX
	sliceY
	track
	n
	figure
	axes3d
	axes2d
	axesX
	axesY
	artists

Detailed Description

Definition at line 101 of file optimizerDisplay.py.

Constructor & Destructor Documentation

__init__()

def lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.__init__	(		self,
			parent,
			xDim,
			yDim,
			n = 0
	)

Definition at line 103 of file optimizerDisplay.py.

    def __init__(self, parent, xDim, yDim, n=0):
        self.parent = parent
        self.xDim = xDim
        self.yDim = yDim
        self.j = self.parent.dimensions.index(self.xDim)
        self.i = self.parent.dimensions.index(self.yDim)
        self.yKey = self.parent.recorder.parameters[self.i]
        self.xKey = self.parent.recorder.parameters[self.j]
        self.zKey = self.parent.recorder.objective
        # grid slice indices corresponding to the dimensions we're plotting
        self.slice2d = [s//2 for s in self.parent.unitGrid.shape[:-1]]
        self.slice2d[self.i] = slice(None)
        self.slice2d[self.j] = slice(None)
        self.slice2d = tuple(self.slice2d)
        self.sliceX = [s//2 for s in self.parent.unitGrid.shape[:-1]]
        self.sliceX[self.j] = slice(None)
        self.sliceX = tuple(self.sliceX)
        self.sliceY = [s//2 for s in self.parent.unitGrid.shape[:-1]]
        self.sliceY[self.i] = slice(None)
        self.sliceY = tuple(self.sliceY)
        self.track = dict(
            x=numpy.array([iteration.sample.get(self.xKey) for iteration in self.parent.track]),
            y=numpy.array([iteration.sample.get(self.yKey) for iteration in self.parent.track]),
            z=numpy.array([iteration.sample.get(self.zKey) for iteration in self.parent.track]),
        )
        self.n = n
        self.figure = matplotlib.pyplot.figure(f"{xDim} vs {yDim}", figsize=(16, 8))
        self.figure.subplots_adjust(left=0.025, right=0.975, bottom=0.08, top=0.95, wspace=0.12)
        self.axes3d = self.figure.add_subplot(1, 2, 1, projection='3d')
        self.axes3d.autoscale(False)
        self.axes3d.set_xlabel(self.xDim)
        self.axes3d.set_ylabel(self.yDim)
        self.axes2d = self.figure.add_subplot(1, 2, 2)
        self.axes2d.set_xlabel(self.xDim)
        self.axes2d.set_ylabel(self.yDim)
        self.axes2d.autoscale(False)
        divider = make_axes_locatable(self.axes2d)
        self.axesX = divider.append_axes("top", 1.5, pad=0.1, sharex=self.axes2d)
        self.axesX.autoscale(False, axis='x')
        hide_xticklabels(self.axesX)
        self.axesY = divider.append_axes("right", 1.5, pad=0.1, sharey=self.axes2d)
        self.axesY.autoscale(False, axis='y')
        hide_yticklabels(self.axesY)
        self.artists = []
        self.guessExtent()
        self.plotTrack()
        self.plotRejected()
        self.plotSurfaces()
 

Member Function Documentation

guessExtent()

def lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.guessExtent

(

self

)

Definition at line 164 of file optimizerDisplay.py.

    def guessExtent(self):
        current = self.parent.track[self.n]
        x = current.sample.get(self.xKey)
        y = current.sample.get(self.yKey)
        zMin1 = current.objectiveValues[self.slice2d].min()
        zMax1 = current.objectiveValues[self.slice2d].max()
        zMin2 = current.objectiveModel[self.slice2d].min()
        zMax2 = current.objectiveModel[self.slice2d].max()
        self.setExtent(x0=x - current.trust, x1=x + current.trust,
                       y0=y - current.trust, y1=y + current.trust,
                       z0=min(zMin1, zMin2), z1=max(zMax1, zMax2), lock=False)
 

move()

def lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.move	(		self,
			n
	)

Definition at line 271 of file optimizerDisplay.py.

    def move(self, n):
        self.n = n
        if not self._lock:
            self.guessExtent()
        for artist in self.artists:
            try:
                artist.remove()
            except TypeError:
                # sometimes matplotlib throws an exception even though everything worked fine
                pass
        self.artists = []
        self.plotSurfaces()
        self.plotRejected()
        self.figure.canvas.draw()

plotRejected()

def lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.plotRejected

(

self

)

Definition at line 221 of file optimizerDisplay.py.

    def plotRejected(self):
        kwds = dict(markeredgewidth=0, markerfacecolor='r', color='r', marker='v')
        current = self.parent.track[self.n]
        cx = current.sample.get(self.xKey)
        cy = current.sample.get(self.yKey)
        cz = current.sample.get(self.zKey)
        for r in current.rejected:
            x = [cx, r.get(self.xKey)]
            y = [cy, r.get(self.yKey)]
            z = [cz, r.get(self.zKey)]
            self.artists.extend(self.axes3d.plot(x, y, z, **kwds))
            self.artists.extend(self.axes2d.plot(x, y, **kwds))
            self.artists.extend(self.axesX.plot(x, z, **kwds))
            self.artists.extend(self.axesY.plot(z, y, **kwds))
 

plotSurfaces()

def lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.plotSurfaces

(

self

)

Definition at line 236 of file optimizerDisplay.py.

    def plotSurfaces(self):
        current = self.parent.track[self.n]
 
        # Start with 2-d and 3-d surfaces
        x = current.grid[self.slice2d + (self.j,)]
        y = current.grid[self.slice2d + (self.i,)]
        z1 = current.objectiveValues[self.slice2d].copy()
        z2 = current.objectiveModel[self.slice2d].copy()
        norm = matplotlib.colors.Normalize(vmin=self.zlim[0], vmax=self.zlim[1])
 
        self._contour(self.axes2d, x, y, z1, cmap=matplotlib.cm.spring, norm=norm)
        self._contour(self.axes2d, x, y, z2, cmap=matplotlib.cm.winter, norm=norm)
 
        # matplotlib doesn't do clipping in 3d, so we'll do that manually
        if self._clipZ(x, y, z1):
            self._contour(self.axes3d, x, y, z1, cmap=matplotlib.cm.spring, norm=norm)
            self.artists.append(self.axes3d.plot_surface(x, y, z1, rstride=1, cstride=1,
                                                         cmap=matplotlib.cm.spring, norm=norm,
                                                         linewidth=0, antialiased=1, alpha=0.5))
        if self._clipZ(x, y, z2):
            self._contour(self.axes3d, x, y, z2, cmap=matplotlib.cm.winter, norm=norm)
            self.artists.append(self.axes3d.plot_surface(x, y, z2, rstride=1, cstride=1,
                                                         cmap=matplotlib.cm.winter, norm=norm,
                                                         linewidth=0, antialiased=1, alpha=0.5))
 
        # Now the 1-d surfaces
        self.artists.extend(self.axesX.plot(current.grid[self.sliceX + (self.j,)],
                                            current.objectiveValues[self.sliceX], 'm-'))
        self.artists.extend(self.axesX.plot(current.grid[self.sliceX + (self.j,)],
                                            current.objectiveModel[self.sliceX], 'c-'))
        self.artists.extend(self.axesY.plot(current.objectiveValues[self.sliceY],
                                            current.grid[self.sliceY + (self.i,)], 'm-'))
        self.artists.extend(self.axesY.plot(current.objectiveModel[self.sliceY],
                                            current.grid[self.sliceY + (self.i,)], 'c-'))
 

plotTrack()

def lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.plotTrack

(

self

)

Definition at line 214 of file optimizerDisplay.py.

    def plotTrack(self):
        kwds = dict(markeredgewidth=0, markerfacecolor='g', color='g', marker='o')
        self.axes3d.plot(self.track['x'], self.track['y'], self.track['z'], **kwds)
        self.axes2d.plot(self.track['x'], self.track['y'], **kwds)
        self.axesX.plot(self.track['x'], self.track['z'], **kwds)
        self.axesY.plot(self.track['z'], self.track['y'], **kwds)
 

setExtent()

def lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.setExtent	(		self,
			x0 = None,
			x1 = None,
			y0 = None,
			y1 = None,
			z0 = None,
			z1 = None,
			lock = True
	)

Definition at line 176 of file optimizerDisplay.py.

    def setExtent(self, x0=None, x1=None, y0=None, y1=None, z0=None, z1=None, lock=True):
        if x0 is None:
            x0 = self._extent[0]
        if x1 is None:
            x1 = self._extent[1]
        if y0 is None:
            y0 = self._extent[2]
        if y1 is None:
            y1 = self._extent[3]
        if z0 is None:
            z0 = self._extent[4]
        if z1 is None:
            z1 = self._extent[5]
        self._extent = (x0, x1, y0, y1, z0, z1)
        self._lock = lock
        self.axes3d.set_xlim(*self.xlim)
        self.axes3d.set_ylim(*self.ylim)
        self.axes3d.set_zlim(*self.zlim)
        self.axes2d.set_xlim(*self.xlim)
        self.axes2d.set_ylim(*self.ylim)
        self.axesX.set_ylim(*self.zlim)
        self.axesY.set_xlim(*self.zlim)
 

xlim()

def lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.xlim

(

self

)

Definition at line 153 of file optimizerDisplay.py.

    def xlim(self):
        return self._extent[:2]
 

ylim()

def lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.ylim

(

self

)

Definition at line 157 of file optimizerDisplay.py.

    def ylim(self):
        return self._extent[2:4]
 

zlim()

def lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.zlim

(

self

)

Definition at line 161 of file optimizerDisplay.py.

    def zlim(self):
        return self._extent[4:]
 

Member Data Documentation

artists

lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.artists

Definition at line 146 of file optimizerDisplay.py.

axes2d

lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.axes2d

Definition at line 135 of file optimizerDisplay.py.

axes3d

lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.axes3d

Definition at line 131 of file optimizerDisplay.py.

axesX

lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.axesX

Definition at line 140 of file optimizerDisplay.py.

axesY

lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.axesY

Definition at line 143 of file optimizerDisplay.py.

figure

lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.figure

Definition at line 129 of file optimizerDisplay.py.

i

lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.i

Definition at line 108 of file optimizerDisplay.py.

j

lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.j

Definition at line 107 of file optimizerDisplay.py.

n

lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.n

Definition at line 128 of file optimizerDisplay.py.

parent

lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.parent

Definition at line 104 of file optimizerDisplay.py.

slice2d

lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.slice2d

Definition at line 113 of file optimizerDisplay.py.

sliceX

lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.sliceX

Definition at line 117 of file optimizerDisplay.py.

sliceY

lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.sliceY

Definition at line 120 of file optimizerDisplay.py.

track

lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.track

Definition at line 123 of file optimizerDisplay.py.

xDim

lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.xDim

Definition at line 105 of file optimizerDisplay.py.

xKey

lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.xKey

Definition at line 110 of file optimizerDisplay.py.

yDim

lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.yDim

Definition at line 106 of file optimizerDisplay.py.

yKey

lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.yKey

Definition at line 109 of file optimizerDisplay.py.

zKey

lsst.meas.modelfit.display.optimizerDisplay.OptimizerDisplayFigure.zKey

Definition at line 111 of file optimizerDisplay.py.

---

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

• /j/snowflake/release/lsstsw/stack/lsst-scipipe-0.7.0/Linux64/meas_modelfit/22.0.1-4-g44f2e3d+9e4ab0f4fa/python/lsst/meas/modelfit/display/optimizerDisplay.py