Accenture / AmpliGraph

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

ConvE has no attribute tensorboard_logs_path #264

Open ramesesz opened 2 years ago

ramesesz commented 2 years ago

When training ConvE with early stopping, the following error gets thrown:

Traceback (most recent call last): File "ConvE(1-n)_fb15k.py", line 47, in early_stopping_params=early_stopping_params File "/home/ramab/anaconda3/envs/ampligraph/lib/python3.7/site-packages/ampligraph/latent_features/models/ConvE.py", line 753, in fit raise e File "/home/ramab/anaconda3/envs/ampligraph/lib/python3.7/site-packages/ampligraph/latent_features/models/ConvE.py", line 744, in fit if self._perform_early_stopping_test(epoch): File "/home/ramab/anaconda3/envs/ampligraph/lib/python3.7/site-packages/ampligraph/latent_features/models/EmbeddingModel.py", line 832, in _perform_early_stopping_test if self.tensorboard_logs_path is not None: AttributeError: 'ConvE' object has no attribute 'tensorboard_logs_path'

The attribute seems to be missing from ConvE, not allowing it to be trained with early stopping regardless of parameter choice.