Skip to content

Commit 01a246c

Browse files
committed
Use undirected Graph projections for all node embeddings
1 parent df426b9 commit 01a246c

File tree

1 file changed

+5
-17
lines changed

1 file changed

+5
-17
lines changed

scripts/reports/NodeEmbeddingsCsv.sh

Lines changed: 5 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ echo "nodeEmbeddingsCsv: CYPHER_DIR=${CYPHER_DIR}"
3232
# Define functions to execute a cypher query from within the given file (first and only argument)
3333
source "${SCRIPTS_DIR}/executeQueryFunctions.sh"
3434

35-
# Define functions to create and delete Graph Projections like "createDirectedDependencyProjection"
35+
# Define functions to create and delete Graph Projections like "createUndirectedDependencyProjection"
3636
source "${SCRIPTS_DIR}/projectionFunctions.sh"
3737

3838
# Create report directory
@@ -145,7 +145,6 @@ nodeEmbeddingsWithNode2Vec() {
145145
# -- Java Artifact Node Embeddings -------------------------------
146146

147147
ARTIFACT_PROJECTION="dependencies_projection=artifact-embeddings"
148-
ARTIFACT_PROJECTION_DIRECTED="dependencies_projection=artifact-directed-embeddings"
149148
ARTIFACT_NODE="dependencies_projection_node=Artifact"
150149
ARTIFACT_WEIGHT="dependencies_projection_weight_property=weight"
151150
ARTIFACT_DIMENSIONS="dependencies_projection_embedding_dimension=16"
@@ -154,15 +153,12 @@ ARTIFACT_DIMENSIONS_HASHGNN="dependencies_projection_embedding_dimension=32"
154153
if createUndirectedDependencyProjection "${ARTIFACT_PROJECTION}" "${ARTIFACT_NODE}" "${ARTIFACT_WEIGHT}"; then
155154
time nodeEmbeddingsWithFastRandomProjection "${ARTIFACT_PROJECTION}" "${ARTIFACT_NODE}" "${ARTIFACT_WEIGHT}" "${ARTIFACT_DIMENSIONS}"
156155
time nodeEmbeddingsWithHashGNN "${ARTIFACT_PROJECTION}" "${ARTIFACT_NODE}" "${ARTIFACT_WEIGHT}" "${ARTIFACT_DIMENSIONS_HASHGNN}"
157-
158-
createDirectedDependencyProjection "${ARTIFACT_PROJECTION_DIRECTED}" "${ARTIFACT_NODE}" "${ARTIFACT_WEIGHT}"
159-
time nodeEmbeddingsWithNode2Vec "${ARTIFACT_PROJECTION_DIRECTED}" "${ARTIFACT_NODE}" "${ARTIFACT_WEIGHT}" "${ARTIFACT_DIMENSIONS}"
156+
time nodeEmbeddingsWithNode2Vec "${ARTIFACT_PROJECTION}" "${ARTIFACT_NODE}" "${ARTIFACT_WEIGHT}" "${ARTIFACT_DIMENSIONS}"
160157
fi
161158

162159
# -- Java Package Node Embeddings --------------------------------
163160

164161
PACKAGE_PROJECTION="dependencies_projection=package-embeddings"
165-
PACKAGE_PROJECTION_DIRECTED="dependencies_projection=package-directed-embeddings"
166162
PACKAGE_NODE="dependencies_projection_node=Package"
167163
PACKAGE_WEIGHT="dependencies_projection_weight_property=weight25PercentInterfaces"
168164
PACKAGE_DIMENSIONS="dependencies_projection_embedding_dimension=32"
@@ -171,15 +167,12 @@ PACKAGE_DIMENSIONS_HASHGNN="dependencies_projection_embedding_dimension=64"
171167
if createUndirectedDependencyProjection "${PACKAGE_PROJECTION}" "${PACKAGE_NODE}" "${PACKAGE_WEIGHT}"; then
172168
time nodeEmbeddingsWithFastRandomProjection "${PACKAGE_PROJECTION}" "${PACKAGE_NODE}" "${PACKAGE_WEIGHT}" "${PACKAGE_DIMENSIONS}"
173169
time nodeEmbeddingsWithHashGNN "${PACKAGE_PROJECTION}" "${PACKAGE_NODE}" "${PACKAGE_WEIGHT}" "${PACKAGE_DIMENSIONS_HASHGNN}"
174-
175-
createDirectedDependencyProjection "${PACKAGE_PROJECTION_DIRECTED}" "${PACKAGE_NODE}" "${PACKAGE_WEIGHT}"
176-
time nodeEmbeddingsWithNode2Vec "${PACKAGE_PROJECTION_DIRECTED}" "${PACKAGE_NODE}" "${PACKAGE_WEIGHT}" "${PACKAGE_DIMENSIONS}"
170+
time nodeEmbeddingsWithNode2Vec "${PACKAGE_PROJECTION}" "${PACKAGE_NODE}" "${PACKAGE_WEIGHT}" "${PACKAGE_DIMENSIONS}"
177171
fi
178172

179173
# -- Java Type Node Embeddings -----------------------------------
180174

181175
TYPE_PROJECTION="dependencies_projection=type-embeddings"
182-
TYPE_PROJECTION_DIRECTED="dependencies_projection=type-directed-embeddings"
183176
TYPE_NODE="dependencies_projection_node=Type"
184177
TYPE_WEIGHT="dependencies_projection_weight_property=weight"
185178
TYPE_DIMENSIONS="dependencies_projection_embedding_dimension=64"
@@ -188,15 +181,12 @@ TYPE_DIMENSIONS_HASHGNN="dependencies_projection_embedding_dimension=128"
188181
if createUndirectedJavaTypeDependencyProjection "${TYPE_PROJECTION}"; then
189182
time nodeEmbeddingsWithFastRandomProjection "${TYPE_PROJECTION}" "${TYPE_NODE}" "${TYPE_WEIGHT}" "${TYPE_DIMENSIONS}"
190183
time nodeEmbeddingsWithHashGNN "${TYPE_PROJECTION}" "${TYPE_NODE}" "${TYPE_WEIGHT}" "${TYPE_DIMENSIONS_HASHGNN}"
191-
192-
createDirectedJavaTypeDependencyProjection "${TYPE_PROJECTION_DIRECTED}"
193-
time nodeEmbeddingsWithNode2Vec "${TYPE_PROJECTION_DIRECTED}" "${TYPE_NODE}" "${TYPE_WEIGHT}" "${TYPE_DIMENSIONS}"
184+
time nodeEmbeddingsWithNode2Vec "${TYPE_PROJECTION}" "${TYPE_NODE}" "${TYPE_WEIGHT}" "${TYPE_DIMENSIONS}"
194185
fi
195186

196187
# -- Typescript Module Node Embeddings ---------------------------
197188

198189
MODULE_PROJECTION="dependencies_projection=typescript-module-embeddings"
199-
MODULE_PROJECTION_DIRECTED="dependencies_projection=typescript-module-directed-embeddings"
200190
MODULE_NODE="dependencies_projection_node=Module"
201191
MODULE_WEIGHT="dependencies_projection_weight_property=lowCouplingElement25PercentWeight"
202192
MODULE_DIMENSIONS="dependencies_projection_embedding_dimension=32"
@@ -205,9 +195,7 @@ MODULE_DIMENSIONS_HASHGNN="dependencies_projection_embedding_dimension=64"
205195
if createUndirectedDependencyProjection "${MODULE_PROJECTION}" "${MODULE_NODE}" "${MODULE_WEIGHT}"; then
206196
time nodeEmbeddingsWithFastRandomProjection "${MODULE_PROJECTION}" "${MODULE_NODE}" "${MODULE_WEIGHT}" "${MODULE_DIMENSIONS}"
207197
time nodeEmbeddingsWithHashGNN "${MODULE_PROJECTION}" "${MODULE_NODE}" "${MODULE_WEIGHT}" "${MODULE_DIMENSIONS_HASHGNN}"
208-
209-
createDirectedDependencyProjection "${MODULE_PROJECTION_DIRECTED}" "${MODULE_NODE}" "${MODULE_WEIGHT}"
210-
time nodeEmbeddingsWithNode2Vec "${MODULE_PROJECTION_DIRECTED}" "${MODULE_NODE}" "${MODULE_WEIGHT}" "${MODULE_DIMENSIONS}"
198+
time nodeEmbeddingsWithNode2Vec "${MODULE_PROJECTION}" "${MODULE_NODE}" "${MODULE_WEIGHT}" "${MODULE_DIMENSIONS}"
211199
fi
212200

213201
# ---------------------------------------------------------------

0 commit comments

Comments
 (0)