单选

class SingleSelection(num_anchors: int = 32)[source]

基础类: AnchorSelection, ABC

单步选择。

初始化策略。

Parameters:

num_anchors (int) – 选择锚节点的数量。 # TODO: 允许相对

方法总结

__call__(edge_index[, known_anchors])

选择锚节点。

rank(edge_index)

对节点进行排序。

方法文档

__call__(edge_index: ndarray, known_anchors: ndarray | None = None) ndarray[source]

选择锚节点。

注意

如果边索引中存在的实体较少,则所选锚点的数量可能小于\(k\)

Parameters:
  • edge_index (ndarray) – 形状: (m, 2) edge_index,即邻接列表。

  • known_anchors (ndarray | None) – numpy.ndarray 一个已知锚点的数组,用于获取唯一的锚点

Returns:

(k,) 选中的实体ID

Return type:

ndarray

abstract rank(edge_index: ndarray) ndarray[来源]

对节点进行排序。

Parameters:

edge_index (ndarray) – 形状: (m, 2) edge_index,即邻接列表。

Returns:

(n,) 节点ID按锚点选择偏好降序排列。

Return type:

ndarray