Head's Up! These forums are read-only. All users and content have migrated. Please join us at community.neo4j.com.
05-24-2021 11:01 PM
I am testing graph embedding algorithm, especially fastRP. My split all nodes in my graph into test and train sets, and then evaluate fastRP/fastExtendedRP's prediction capability. I am comparing performances in the 3 categories:
In all cases, I am running the classification algorithm:
CALL gds.alpha.ml.nodeClassification.train
The results are below:
It seems that the structure of my graph doesn't help too much, while bert embedding alone is far better than fastRP or fastExtendedRP. Since fastExtendedRP has taken advantage of the bertEmbedding, I don't expect it is significantly worse than bertEmbedding alone. I guess it might be due to my parameter setting when training it, and those are my settings:
CALL gds.beta.fastRPExtended.write(
'nodeGraph',
{ embeddingDimension: 512,
iterationWeights: [0.0, 1.0, 1.0, 1.0],
normalizationStrength: 0,
propertyDimension: 96,
featureProperties: ['bertEmbedding'],
writeProperty: 'graphEmbedding'
}
)
YIELD nodePropertiesWritten
""")
Graph embedding authors, any suggestions on tuning in the parameters? The BERT embedding is 768 dimension in standard form.
All the sessions of the conference are now available online