Skip to content

Commit 39f0aa0

Browse files
committed
Added doc about triplet scoring
1 parent bb1cd0a commit 39f0aa0

File tree

1 file changed

+14
-5
lines changed

1 file changed

+14
-5
lines changed

examples/kge-distmult-nations.ipynb

Lines changed: 14 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -228,6 +228,16 @@
228228
"G_train, G_valid, G_test = project_graphs()"
229229
]
230230
},
231+
{
232+
"cell_type": "code",
233+
"execution_count": null,
234+
"id": "21da1ea76d247803",
235+
"metadata": {},
236+
"outputs": [],
237+
"source": [
238+
"G_train.relationship_types()"
239+
]
240+
},
231241
{
232242
"cell_type": "code",
233243
"execution_count": null,
@@ -242,11 +252,10 @@
242252
"gds.kge.model.train(\n",
243253
" G_train,\n",
244254
" model_name=model_name,\n",
245-
" scoring_function=\"distmult\",\n",
246-
" num_epochs=1,\n",
247-
" embedding_dimension=10,\n",
248-
" epochs_per_checkpoint=0,\n",
249-
" epochs_per_val=0,\n",
255+
" scoring_function=\"transe\",\n",
256+
" num_epochs=30,\n",
257+
" embedding_dimension=64,\n",
258+
" split_ratios={\"TRAIN\": 0.8, \"VALID\": 0.1, \"TEST\": 0.1},\n",
250259
")\n",
251260
"\n",
252261
"predict_result = gds.kge.model.predict(\n",

0 commit comments

Comments
 (0)