diff --git a/cornac/models/bivaecf/bivae.py b/cornac/models/bivaecf/bivae.py index 5221ca813c0862133767f276530be1f501ec6688..f8fe7154d47e4651185863970eaf67ae31570bfc 100644 --- a/cornac/models/bivaecf/bivae.py +++ b/cornac/models/bivaecf/bivae.py @@ -14,6 +14,7 @@ # ============================================================================ import itertools as it +import logging import numpy as np import torch @@ -290,7 +291,7 @@ def learn( else: patience_counter += 1 if patience_counter >= patience: - print(f"Early stopping at epoch {epoch} due to no improvement in validation loss.") + logging.info(f"Early stopping at epoch {epoch} due to no improvement in validation loss.") break # Stop training # infer mu_beta