Accenture / AmpliGraph

Python library for Representation Learning on Knowledge Graphs https://docs.ampligraph.org
Apache License 2.0
2.12k stars 252 forks source link

Feature/230 #244

Closed adrijanik closed 3 years ago

adrijanik commented 3 years ago

Related Issue(s)

Issue #230

Description of Changes

Added parameter tensorboard=False to model fit function (EmbeddingModel derivatives) to visualize average training loss after each epoch with tensorboard. After model is trained with e.g. CompleX(...,tensorboard=True), tensorboard_files folder is created where loss is being logged. To see the loss, run: tensorboard --logdir tensorboard_files/ and go to your browser to see the loss plotted.

Any other comments?