Skip to main content

图形工具

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: 代理人的列表

检查以下内容: 错误

  1. 字典必须以键值对的结构存在。
  2. 每个键必须存在于代理人列表中。
  3. 每个值必须是代理人的列表(而不是字符串)。

警告

  1. 如果存在孤立的代理人节点,则发出警告。
  2. 如果 allowed_speaker_transitions 中的代理人集合与代理人列表不匹配,则发出警告。
  3. 如果 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。