restore_model

ampligraph.latent_features.restore_model(loc)

Restore a saved model from disk.

Examples

>>> from ampligraph.latent_features import restore_model
>>> import numpy as np
>>> EXAMPLE_LOC = 'saved_models' # Assuming that the model is present at this location
>>> restored_model = restore_model(EXAMPLE_LOC)
>>> y_pred_after = restored_model.predict(np.array([['f', 'y', 'e'], ['b', 'y', 'd']]))
>>> print(y_pred_after)
[1.261404, -1.324778]

Parameters:loc (string) – Directory containing the saved model Returns:model – a neural knowledge graph embedding model Return type:trained model