search

neo4j-graph-analytics / ml-models

Machine Learning Procedures and Functions for Neo4j
https://github.com/neo4j-graph-analytics/ml-models/releases/tag/1.0.0
Apache License 2.0
64 stars 23 forks source link

Embedding size with embedding.deepgl #18

Open akastrin opened 4 years ago

akastrin commented 4 years ago

Hi,

I try to figure out how to properly set the number of desired features with embedding.deepgl algorithms. Currently, my setup goes as follows:

CALL
embedding.deepgl(
    null,
    null,
    {
         pruningLambda: 0.6,
         diffusions: 3,
         iterations: 3,
         embeddingSize: 128
    }
)

I explicitely set embedding size to 128, however the resulting size is 158. Could somebody please explain why?

Thanks, Andrej