<< Click to Display Table of Contents >> Navigation:  Tutorials > Tutorial 3: Exploring the contents of a model > Tutorial3.py

import pysmile

 

# Tutorial3 loads the XDSL file and prints the information

# about the structure (nodes and arcs) and the parameters 

# (conditional probabilities of the nodes) of the network.

 

class Tutorial3:

    def __init__(self):

        print("Starting Tutorial3...")

        net = pysmile.Network()

        

        # load the network created by Tutorial1

        net.read_file("tutorial1.xdsl")

        for h in net.get_all_nodes():

            self.print_node_info(net, h)

        print("Tutorial3 complete.")

 

    def print_node_info(self, net, node_handle):

        print(f"Node id/name: {net.get_node_id(node_handle)}/{net.get_node_name(node_handle)}")

        print(f"  Outcomes: {' '.join(net.get_outcome_ids(node_handle))}")

       

        parent_ids = net.get_parent_ids(node_handle)

        if len(parent_ids) > 0:

            print(f"  Parents: {' '.join(parent_ids)}")

        child_ids = net.get_child_ids(node_handle)

        if len(child_ids) > 0:

            print(f"  Children: {' '.join(child_ids)}")

        

        self.print_cpt_matrix(net, node_handle)

 

    def print_cpt_matrix(self, net, node_handle):

        cpt = net.get_node_definition(node_handle)

        parents = net.get_parents(node_handle)

        dim_count = 1 + len(parents)

        

        dim_sizes = [0] * dim_count

        for i in range(0, dim_count - 1):

            dim_sizes[i] = net.get_outcome_count(parents[i])

        dim_sizes[len(dim_sizes) - 1] = net.get_outcome_count(node_handle)

        

        coords = [0] * dim_count

        for elem_idx in range(0, len(cpt)):

            self.index_to_coords(elem_idx, dim_sizes, coords)

            

            outcome = net.get_outcome_id(node_handle, coords[dim_count - 1])

            print(f"    P({outcome}", end="")

            

            if dim_count > 1:

                print(" | ", end="")

                for parent_idx in range(0, len(parents)):

                    if parent_idx > 0: 

                        print(",", end="")

                    parent_handle = parents[parent_idx]

                    print(f"{net.get_node_id(parent_handle)}=" 

                        + f"{net.get_outcome_id(parent_handle, coords[parent_idx])}", end="")

            

            prob = cpt[elem_idx]

            print(f")={prob}")

 

    def index_to_coords(self, index, dim_sizes, coords):

        prod = 1

        for i in range(len(dim_sizes) - 1, -1, -1):

            coords[i] = int(index / prod) % dim_sizes[i]

            prod *= dim_sizes[i]