velocitatem/PHANTOM
1
0
Fork 0
mirror of https://github.com/velocitatem/PHANTOM.git synced 2026-05-31 16:43:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

featuer: dot exporter

Browse Source
This commit is contained in:
Daniel Rosel
2026-01-10 10:33:56 +01:00
parent b1882b6049
commit c56c7f6537
1 changed files with 9 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
sim/rl/behavior_loader/models.py
View File

@@ -85,7 +85,7 @@ class BehaviorModel:
path.append(curr) path.append(curr)
return path return path
def visualize_mdp(model: BehaviorModel, threshold: float = 0.05, output: str = "mdp_graph", fmt: str = "svg", view: bool = False): def visualize_mdp(model: BehaviorModel, threshold: float = 0.05, output: str = "mdp_graph", fmt: str = "svg", view: bool = False, export_dot: bool = False):
"""visualize MDP as directed graph using graphviz, aggregated by event type""" """visualize MDP as directed graph using graphviz, aggregated by event type"""
if not model.mdp: raise ValueError("build MDP first") if not model.mdp: raise ValueError("build MDP first")
@@ -125,6 +125,13 @@ def visualize_mdp(model: BehaviorModel, threshold: float = 0.05, output: str = "
g.render(output, view=view, cleanup=True) g.render(output, view=view, cleanup=True)
print(f"Saved MDP graph to {output}.{fmt}") print(f"Saved MDP graph to {output}.{fmt}")
if export_dot:
dot_file = f"{output}.dot"
with open(dot_file, 'w') as f:
f.write(g.source)
print(f"Exported DOT source to {dot_file}")
return g return g
if __name__ == "__main__": if __name__ == "__main__":
@@ -134,4 +141,4 @@ if __name__ == "__main__":
if not mdp['states']: if not mdp['states']:
print("No states found") print("No states found")
exit(1) exit(1)
visualize_mdp(model, threshold=0.05, output="mdp_viz", fmt="svg") visualize_mdp(model, threshold=0.05, output="mdp_viz", fmt="svg", export_dot=True)