gds.beta.pipeline.nodeClassification.train:找不到目标属性

问题描述 投票:0回答:1

使用 Neo4j v4.4 和 GDS 2.0。我正在尝试训练一个模型。当我输入:

CALL gds.beta.pipeline.nodeClassification.train('individual-graph', {
   pipeline: 'pipe',
   nodeLabels: ['PERSON'],
   modelName: 'xmen-model-fastRP',
   targetProperty: 'is_risky',
   metrics: ['F1_WEIGHTED','ACCURACY'],
   randomSeed: 2
   }) YIELD modelInfo
  RETURN
  modelInfo.bestParameters AS winningModel,
  modelInfo.metrics.F1_WEIGHTED.outerTrain AS trainGraphScore,
  modelInfo.metrics.F1_WEIGHTED.test AS testGraphScore

我收到以下错误消息:

调用过程失败

gds.beta.pipeline.nodeClassification.train
:
原因:
java.lang.IllegalArgumentException
:在具有节点属性的图表中找不到目标属性
is_risky
:[[嵌入]]

我做错了什么?你能帮忙吗?

neo4j
1个回答
0
投票

这意味着在节点 PERSON 中没有找到属性“is_risky”。唯一现有的属性是嵌入。

查看 neo4j 文档中的示例(https://neo4j.com/docs/graph-data-science/current/machine-learning/nodeclassification-pipelines/)将使您了解错误是什么。以下是您遇到的类似问题的示例。

Target property `my_class` not found in graph 
with node properties: [[sizePerStory, class], [sizePerStory]]

如您所见,算法将为您提供可用于预测的属性列表。

© www.soinside.com 2019 - 2024. All rights reserved.