Old engine for Continuous Time Bayesian Networks. Superseded by reCTBN. 🐍
https://github.com/madlabunimib/PyCTBN
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
46 lines
1.3 KiB
46 lines
1.3 KiB
from enum import Enum
|
|
|
|
class Color(Enum):
|
|
WHITE = 0
|
|
GRAY = 1
|
|
BLACK = 2
|
|
|
|
class Node():
|
|
"""
|
|
Astrae il concetto di nodo appartenente ad un grafo G.
|
|
Un nodo viene univocamente identificato dal suo state_id, il tag node_id è utilizzato per stabilire la posizione
|
|
del nodo stesso nel grafo rappresentato attraverso la matrice di adiacenza.
|
|
:state_id: UID del nodo
|
|
:node_id: int
|
|
:color: indica che il nodo sia stato visitato o meno durante la successiva BFS visit
|
|
"""
|
|
|
|
def __init__(self, state_id, node_id=-1, color=Color.WHITE):
|
|
self.state_id = state_id
|
|
self.node_id = node_id
|
|
self.color = color
|
|
|
|
def __key(self):
|
|
return (self.state_id)
|
|
|
|
def __hash__(self):
|
|
return hash(self.__key())
|
|
|
|
|
|
def __eq__(self, other):
|
|
"""
|
|
Controlla l'uguaglianza di due oggetti Node.
|
|
Due nodi sono uguali sse hanno lo stesso state_id
|
|
Parameters:
|
|
other: oggetto nodo con cui effettuare il confronto
|
|
Returns:
|
|
boolean
|
|
"""
|
|
if isinstance(other, Node):
|
|
return self.__key() == other.__key()
|
|
return NotImplemented
|
|
|
|
def __repr__(self):
|
|
return "<%s, %d, %s>"% (self.state_id, self.node_id, self.color)
|
|
|
|
|
|
|