mirror of
https://github.com/SheffieldML/GPy.git
synced 2026-05-08 11:32:39 +02:00
[pydot] build pydot with new observer list
This commit is contained in:
parent
f110bbd4c8
commit
851e63476c
3 changed files with 7 additions and 6 deletions
|
|
@ -58,6 +58,7 @@ class ObservablesList(object):
|
||||||
def __repr__(self):
|
def __repr__(self):
|
||||||
return self._poc.__repr__()
|
return self._poc.__repr__()
|
||||||
|
|
||||||
|
|
||||||
def add(self, priority, observable, callble):
|
def add(self, priority, observable, callble):
|
||||||
if observable is not None:
|
if observable is not None:
|
||||||
ins = 0
|
ins = 0
|
||||||
|
|
|
||||||
|
|
@ -57,9 +57,9 @@ class Param(OptimizationHandlable, ObsAr):
|
||||||
|
|
||||||
def build_pydot(self,G):
|
def build_pydot(self,G):
|
||||||
import pydot
|
import pydot
|
||||||
node = pydot.Node(id(self), shape='record', label=self.name)
|
node = pydot.Node(id(self), shape='trapezium', label=self.name)#, fontcolor='white', color='white')
|
||||||
G.add_node(node)
|
G.add_node(node)
|
||||||
for o in self.observers.keys():
|
for _, o, _ in self.observers:
|
||||||
label = o.name if hasattr(o, 'name') else str(o)
|
label = o.name if hasattr(o, 'name') else str(o)
|
||||||
observed_node = pydot.Node(id(o), label=label)
|
observed_node = pydot.Node(id(o), label=label)
|
||||||
G.add_node(observed_node)
|
G.add_node(observed_node)
|
||||||
|
|
|
||||||
|
|
@ -82,15 +82,15 @@ class Parameterized(Parameterizable):
|
||||||
import pydot # @UnresolvedImport
|
import pydot # @UnresolvedImport
|
||||||
iamroot = False
|
iamroot = False
|
||||||
if G is None:
|
if G is None:
|
||||||
G = pydot.Dot(graph_type='digraph')
|
G = pydot.Dot(graph_type='digraph', bgcolor=None)
|
||||||
iamroot=True
|
iamroot=True
|
||||||
node = pydot.Node(id(self), shape='record', label=self.name)
|
node = pydot.Node(id(self), shape='box', label=self.name)#, color='white')
|
||||||
G.add_node(node)
|
G.add_node(node)
|
||||||
for child in self._parameters_:
|
for child in self._parameters_:
|
||||||
child_node = child.build_pydot(G)
|
child_node = child.build_pydot(G)
|
||||||
G.add_edge(pydot.Edge(node, child_node))
|
G.add_edge(pydot.Edge(node, child_node))#, color='white'))
|
||||||
|
|
||||||
for o in self.observers.keys():
|
for _, o, _ in self.observers:
|
||||||
label = o.name if hasattr(o, 'name') else str(o)
|
label = o.name if hasattr(o, 'name') else str(o)
|
||||||
observed_node = pydot.Node(id(o), label=label)
|
observed_node = pydot.Node(id(o), label=label)
|
||||||
G.add_node(observed_node)
|
G.add_node(observed_node)
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue