图形工具
has_self_loops
def has_self_loops(allowed_speaker_transitions: Dict) -> bool
如果 allowed_speaker_transitions_Dict 中存在自环,则返回 True。
check_graph_validity
def check_graph_validity(allowed_speaker_transitions_dict: Dict,
agents: List[Agent])
allowed_speaker_transitions_dict: 一个以键值对形式存在的字典。键是代理人的名称,值是该代理人可以转换到的代理人的名称列表。 agents: 代理人的列表
检查以下内容: 错误
- 字典必须以键值对的结构存在。
- 每个键必须存在于代理人列表中。
- 每个值必须是代理人的列表(而不是字符串)。
警告
- 如果存在孤立的代理人节点,则发出警告。
- 如果 allowed_speaker_transitions 中的代理人集合与代理人列表不匹配,则发出警告。
- 如果 allowed_speaker_transitions_dict 的任何值中存在重复的代理人,则发出警告。
invert_disallowed_to_allowed
def invert_disallowed_to_allowed(disallowed_speaker_transitions_dict: dict,
agents: List[Agent]) -> dict
从完全连接的 allowed_speaker_transitions_dict 开始。根据 disallowed_speaker_transitions_dict 从完全连接的 allowed_speaker_transitions_dict 中删除边,形成 allowed_speaker_transitions_dict。
visualize_speaker_transitions_dict
def visualize_speaker_transitions_dict(speaker_transitions_dict: dict,
agents: List[Agent],
export_path: Optional[str] = None)
使用 networkx 可视化 speaker_transitions_dict。