Skip to content

Commit c9c31b2

Browse files
committed
Enable mlflow
1 parent fca912b commit c9c31b2

File tree

4 files changed

+65
-269
lines changed

4 files changed

+65
-269
lines changed

doc/modules/ROOT/pages/gds-session-algorithms/knowledge-graph-embeddings.adoc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -268,8 +268,8 @@ When all three keys are present, the sum of values must be equal to 1.0.
268268

269269
| epochs_per_val
270270
| int
271-
| 50
272-
| Number of epochs between validations (must be greater than or equal to 0)
271+
| 0
272+
| Number of epochs between validations (must be greater than or equal to 0). If 0, validation is not performed.
273273

274274
| inner_norm
275275
| bool

examples/kge-distmult-nations.ipynb

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -313,6 +313,7 @@
313313
" num_epochs=30,\n",
314314
" embedding_dimension=64,\n",
315315
" split_ratios={\"TRAIN\": 0.8, \"VALID\": 0.1, \"TEST\": 0.1},\n",
316+
" mlflow_experiment_name=\"Nations-train\",\n",
316317
")\n",
317318
"\n",
318319
"predict_result = gds.kge.model.predict(\n",

0 commit comments

Comments
 (0)