def add_boolean_proxy_for(self, label: str, shape: Collection[int] = None) -> Vertex:
"""
Creates a proxy vertex for the given label and adds to the sequence item
"""
if shape is None:
return Vertex._from_java_vertex(self.unwrap().addBooleanProxyFor(_VertexLabel(label).unwrap()))
else:
return Vertex._from_java_vertex(self.unwrap().addBooleanProxyFor(_VertexLabel(label).unwrap(), shape))
def evaluate_and_check_for_increment(model: LambdaModel, v_in: Vertex) -> None:
v_out: Vertex = model.get_double_model_output_vertex("out")
v_in.set_value(1.)
v_out.unwrap().eval()
assert v_out.get_value() == 2.
v_in.set_value(1.1)
v_out.unwrap().eval()
assert v_out.get_value() == 2.1
def get_contents(self) -> Dict[str, Vertex]:
def get_unqualified_name_or_proxy_name(key, vertex) -> str:
if is_instance_of(k._gateway, vertex, "io.improbable.keanu.vertices.ProxyVertex"):
return "proxy_for." + key.getUnqualifiedName()
return key.getUnqualifiedName()
return {
get_unqualified_name_or_proxy_name(k, v): Vertex._from_java_vertex(v)
for k, v in self.unwrap().getContents().items()
}
def create_description(vertex: Vertex) -> str:
return description_creator.createDescription(vertex.unwrap())
def get(self, label: str) -> Vertex:
return Vertex._from_java_vertex(self.unwrap().get(_VertexLabel(label).unwrap()))
def add(self, vertex: Vertex, label: Optional[str] = None) -> None:
if label is None:
self.unwrap().add(vertex.unwrap())
else:
self.unwrap().add(_VertexLabel(label).unwrap(), vertex.unwrap())
def get_acceptance_rate(self, vertex: Vertex) -> float:
return self.unwrap().getAcceptanceRate(vertex.unwrap().getId())