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)

Export embeddings to Tensorboard.

Others

Function to convert a pandas DataFrame with headers into triples.

dataframe_to_triples(X, schema)

Convert DataFrame into triple format.