logging best epoch

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