VizBirth.py 文件源码

def _viz_Gauss(curModel, propModel, Plan,
               curELBO=None, propELBO=None, block=False, **kwargs):
    from ..viz import GaussViz
    from matplotlib import pylab
    pylab.figure()
    h = pylab.subplot(1, 2, 1)
    GaussViz.plotGauss2DFromHModel(curModel, compsToHighlight=Plan['ktarget'])
    h = pylab.subplot(1, 2, 2)
    newCompIDs = np.arange(curModel.obsModel.K, propModel.obsModel.K)
    GaussViz.plotGauss2DFromHModel(propModel, compsToHighlight=newCompIDs)
    pylab.show(block=block)