interactive notebooks

pyAgrum can easily interact with other applications. In this notebook, we propose for example some application tracks with notebook ipywidgets to make the exploration of graphical models and their inferences more interactive.

import pyAgrum as gum
import pyAgrum.lib.notebook as gnb

Listeners and progress bars

import glob
import os.path
from tqdm.auto import tqdm

class TqdmProgressBarLoadListener:
    def __init__(self,filename:str):
        self.pbar=tqdm(total=100,
                      desc=filename,
                      bar_format='{desc}: {percentage:3.0f}%|{bar}|')
    def update(self,progress):
        if progress==200:
            self.pbar.close()
        else:
            self.pbar.update()
            self.pbar.display()


bns={}
for ext in ['dsl','bif']:
    for name in glob.glob(f"res/*.{ext}"):
        progbar=TqdmProgressBarLoadListener(name)
        bns[os.path.basename(name)]=gum.loadBN(name,listeners=[lambda progress:progbar.update(progress)])

Which should give you something like

Animated graphs

ipywidget can be used with different types of objects. Let’s say that you have a class that show the arcs of a Bayesian network only the mutual information of this arc is above a certain threshold:

import pydot as dot

class InformationViewer:
    def __init__(self,bn:gum.BayesNet):
        self.bn=bn

        ie=gum.LazyPropagation(bn)
        self._min=float("inf")
        self._max=float("-inf")
        self._arcs={}
        for x,y in bn.arcs():
            nameX=bn.variable(x).name()
            nameY=bn.variable(y).name()
            ie.addJointTarget({nameX,nameY})
            info=gum.InformationTheory(ie,[nameX],[nameY])
            m=info.mutualInformationXY()
            if self._min>m: self._min=m
            if self._max<m: self._max=m
            self._arcs[x,y]=m

    def min(self):
        return self._min

    def max(self):
        return self._max

    def showBN(self,minVal:float=0):
        graph=dot.Dot(graph_type="digraph",bgcolor="transparent")
        bgcol = gum.config["notebook", "default_node_bgcolor"]
        fgcol = gum.config["notebook", "default_node_fgcolor"]
        for n in self.bn.names():
            graph.add_node(dot.Node('"' + n + '"', style="filled",
                                    fillcolor=bgcol,
                                    fontcolor=fgcol))
        for x,y in self.bn.arcs():
            graph.add_edge(dot.Edge('"' + self.bn.variable(x).name() + '"',
                                    '"' + self.bn.variable(y).name() + '"',
                                    style="invis" if self._arcs[x,y]<minVal else ""))


        size = gum.config["notebook", "default_graph_size"]
        graph.set_size(size)
        return graph

view=InformationViewer(bns['alarm.dsl'])
print(f"min={view.min()} ,max={view.max()}")
gnb.sideBySide(view.showBN(0.3),view.showBN(0.5),
              captions=["BN filtered by $MI>0.3$","BN filtered by $MI>0.5$"])

min=7.940532588376242e-06 ,max=0.8850119269966232

BN filtered by $MI>0.3$

BN filtered by $MI>0.5$

Now we can use this class for animation :

import ipywidgets as widgets
def interactive_view(threshold:float):
    return view.showBN(threshold)
widgets.interact(interactive_view,threshold=(view.min(),
                                             view.max(),
                                             (view.max()-view.min())/100.0));

Which should give you something like

Vizualizing evidence impact

from ipywidgets import interact, fixed

bn = bns['asia.bif']

asia = list(bn.variableFromName("visit_to_Asia").labels())
smoking = list(bn.variableFromName("smoking").labels())
XraY = list(bn.variableFromName("positive_XraY").labels())
cig_ped_day = gum.RangeVariable("cigarettes_per_day","cigarettes_per_day in [0, 10]?",0,10)
bn.add(cig_ped_day)

@interact(bn=fixed(bn), visit_to_Asia=asia, smoking=smoking, positive_XraY=XraY, smoked_cigarettes=(cig_ped_day.minVal(), cig_ped_day.maxVal(), 1))
def evidence_impact(bn, visit_to_Asia, smoking, positive_XraY, smoked_cigarettes):
    evs = {"visit_to_Asia":visit_to_Asia, "smoking":smoking, "positive_XraY":positive_XraY, "cigarettes_per_day":smoked_cigarettes}
    gnb.showInference(bn, evs=evs)