GDS version: 1.7
Neo4j: 4.3.4
CALL gds.beta.graphSage.train(
'nodeGraph',
{
modelName: 'graphSageModel',
aggregator:'pool',
batchSize:32,
activationFunction:'relu',
epochs:3,
searchDepth:3,
sampleSizes:[25,10],
learningRate:0.001,
embeddingDimension:128,
featureProperties:['degree', 'pageRank'],
projectedFeatureDimension: 2,
randomSeed: 46,
concurrency: 4
}
)
YIELD modelInfo as info
RETURN info.name as modelName,
info.metrics.didConverge as didConverge,
info.metrics.ranEpochs as ranEpochs,
info.metrics.epochLosses as epochLosses
{'modelName': None, 'didConverge': False, 'ranEpochs': 3, 'epochLosses': [729.7173010597527, 674.4603356708892, 583.5583915437866]}
As you can see, the modelName is None. How to get the correct name? I didn't see the documentation on what fields the 'modelInfo' has in Neo4j's documentation.