✓ File Writeback ✕ Property Writeback ✓ Direct Return ✓ Stream Return ✕ Stats

Overview

In cosine similarity, data objects in a dataset are treated as vectors, and it uses the cosine value of the angle between two vectors to indicate the similarity between them. In the graph, specifying N numeric properties (features) of nodes to form N-dimensional vectors, two nodes are considered similar if their vectors are similar.

Cosine similarity ranges from -1 to 1; 1 means that the two vectors have the same direction, -1 means that the two vectors have the opposite direction.

In 2-dimensional space, the cosine similarity between vectors A(a₁, a₂) and B(b₁, b₂) is computed as:

In 3-dimensional space, the cosine similarity between vectors A(a₁, a₂, a₃) and B(b₁, b₂, b₃) is computed as:

The following diagram shows the relationship between vectors A and B in 2D and 3D spaces, as well as the angle θ between them:

Generalize to N-dimensional space, the cosine similarity is computed as:

Considerations

• Theoretically, the calculation of cosine similarity between two nodes does not depend on their connectivity.
• The value of cosine similarity is independent of the length of the vectors, but only the direction of the vectors.

Syntax

• Command: algo(similarity)
• Parameters:

This algorithm has two calculation modes:

1. Pairing: when ids/uuids and ids2/uuids2 are both configured, pairing nodes in the first group with nodes in the second group (Cartesian product) to compute pair-wise similarities.
2. Selection: when only ids/uuids is configured, for each node in the group, computing pair-wise similarities between it and all other nodes in the graph in order to select the most similar nodes, the returned top_list includes all nodes that have similarity > 0 with it and is ordered by the descending similarity.

Examples

The example graph has 4 products (edges are ignored), each product has properties price, weight, weight and height:

File Writeback

algo(similarity).params({
  uuids: [1], 
  uuids2: [2,3,4],
  node_schema_property: [price,weight,width,height]
}).write({
  file:{ 
    filename: "cs_result"
  }
})

Results: File cs_result

product1,product2,0.986529
product1,product3,0.878858
product1,product4,0.816876

algo(similarity).params({
  uuids: [1,2,3,4],
  node_schema_property: [price,weight,width,height],
  type: "cosine"
}).write({
  file:{ 
    filename: "list"
  }
})

Results: File list

product1,product2:0.986529;product3:0.878858;product4:0.816876;
product2,product1:0.986529;product3:0.934217;product4:0.881988;
product3,product2:0.934217;product4:0.930153;product1:0.878858;
product4,product3:0.930153;product2:0.881988;product1:0.816876;

Direct Return

algo(similarity).params({
  uuids: [1], 
  uuids2: [2,3,4],
  node_schema_property: [price,weight,width,height],
  type: "cosine"
}) as cs
return cs order by cs.similarity

Results: cs

algo(similarity).params({
  uuids: [1,2],
  type: "cosine",
  node_schema_property: [price,weight,width,height],
  top_limit: 1
}) as top
return top

Results: top

Stream Return

algo(similarity).params({
  uuids: [3], 
  uuids2: [1,2,4],
  node_schema_property: [price,weight,width,height],
  type: "cosine"
}).stream() as cs
where cs.similarity > 0.9 
return cs

Results: cs

algo(similarity).params({
  uuids: [1,3],
  node_schema_property: [price,weight,width,height],
  type: "cosine",
  top_limit: 1
}).stream() as top
return top

Results: top