Skip to content
This repository has been archived by the owner on Nov 28, 2023. It is now read-only.

Commit

Permalink
Update NeuralNet.py
Browse files Browse the repository at this point in the history
  • Loading branch information
manonreau committed Dec 22, 2021
1 parent 0a27faa commit 70041ec
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion deeprank/learn/NeuralNet.py
Original file line number Diff line number Diff line change
Expand Up @@ -502,7 +502,7 @@ def load_nn_params(self):
try:
self.hit_cutoff = self.state['hit_cutoff']
except Exception:
print(f'No "hit_cutoff" found in {self.pretrained_model}. Please set it in function "test()" when doing benchmark"')
logger.warning(f'No "hit_cutoff" found in {self.pretrained_model}. Please set it in function "test()" when doing benchmark"')

def load_data_params(self):
"""Get dataset parameters from a saved model."""
Expand Down

0 comments on commit 70041ec

Please sign in to comment.