GraphSage 模型

class graphdatascience.model.graphsage_model.GraphSageModel

表示模型目录中的 GraphSAGE 模型。使用 gds.beta.graphSage.train() 构建此模型。

creation_time() → Any

获取模型的创建时间。

返回:

模型的创建时间。

返回类型:

Any

drop(failIfMissing: bool = False) → Series[Any]

删除模型。

参数:

failIfMissing (bool) – 如果为 True，则在模型不存在时抛出错误。如果为 False，则不抛出错误。

返回:

删除操作的结果。

返回类型:

Series[Any]

exists() → bool

检查模型是否存在。

返回:

如果模型存在则返回 True，否则返回 False。

返回类型:

bool

graph_schema() → Series[Any]

获取模型的图谱模式 (graph schema)。

返回:

模型的图谱模式。

返回类型:

Series[Any]

loaded() → bool

检查模型是否已加载到内存中。

返回:

如果模型已加载到内存则返回 True，否则返回 False。

返回类型:

bool

metrics() → Series[Any]

获取模型的指标。

返回:

模型的指标。

返回类型:

Series[Any]

model_info() → dict[str, Any]

获取模型信息。

返回:

模型信息。

返回类型:

dict[str, Any]

name() → str

获取模型名称。

返回:

模型名称。

返回类型:

str

predict_mutate(G: Graph, **config: Any) → Series[Any]

使用模型对给定的图进行预测，并将结果写入 (mutate) 图中。

参数:

• G (Graph) – 要进行预测的图。

• **config (Any) – 预测的配置。

返回:

Mutate 操作的结果。

返回类型:

Series[Any]

predict_mutate_estimate(G: Graph, **config: Any) → Series[Any]

估算使用模型在给定图上进行预测所需的内存。

参数:

• G (Graph) – 要进行预测的图。

• **config (Any) – 预测的配置。

返回:

在给定图上进行预测所需的内存。

返回类型:

Series[Any]

predict_stream(G: Graph, **config: Any) → DataFrame

使用模型对给定的图进行预测，并将结果以 DataFrame 形式流式传输。

参数:

• G (Graph) – 要进行预测的图。

• **config (Any) – 预测的配置。

返回:

以 DataFrame 形式返回的预测结果。

返回类型:

DataFrame

predict_stream_estimate(G: Graph, **config:[ Any) → Series[Any]

估算在给定图上进行预测并将结果以 DataFrame 形式流式传输所需的资源。

参数:

• G (Graph) – 要进行预测的图。

• **config (Any) – 预测的配置。

返回:

以 DataFrame 形式返回的预测结果。

返回类型:

Series[Any]

predict_write(G: Graph, **config: Any) → Series[Any]

为给定的图生成嵌入并将结果写入数据库。

参数:

• G (Graph) – 用于生成嵌入的图。

• **config (Any) – 预测的配置。

返回:

写入操作的结果。

返回类型:

Series[Any]

predict_write_estimate(G: Graph, **config: Any) → Series[Any]

估算为给定的图生成嵌入并将结果写入数据库所需的内存。

参数:

• G (Graph) – 用于生成嵌入的图。

• **config (Any) – 预测的配置。

返回:

为给定的图生成嵌入并将结果写入数据库所需的内存。

返回类型:

Series[Any]

published() → bool

检查模型是否已发布。

返回:

如果模型已发布则返回 True，否则返回 False。

返回类型:

bool

shared() → bool

检查模型是否已共享。

返回:

如果模型已共享则返回 True，否则返回 False。

返回类型:

bool

stored() → bool

检查模型是否已存储在磁盘上。

返回:

如果模型已存储在磁盘上则返回 True，否则返回 False。

返回类型:

bool

train_config() → Series[Any]

获取模型的训练配置。

返回:

模型的训练配置。

返回类型:

Series[Any]

type() → str

获取模型类型。

返回:

模型类型。

返回类型:

str