Utils

This module contains utility functions for neural knowledge graph embedding models.

Saving/Restoring Models

Models can be saved and restored from disk. This is useful to avoid re-training a model.

save_model(model[, model_name_path]) Save a trained model to disk.
restore_model([model_name_path]) Restore a saved model from disk.

Visualization

Functions to visualize embeddings.

create_tensorboard_visualizations(model, loc) Create Tensorboard visualization files.