diff --git a/cornac/models/mf/recom_mf.py b/cornac/models/mf/recom_mf.py index 79173ff4a..bcc83e028 100644 --- a/cornac/models/mf/recom_mf.py +++ b/cornac/models/mf/recom_mf.py @@ -114,7 +114,7 @@ def __init__( self.learning_rate = learning_rate self.batch_size = batch_size self.lambda_reg = lambda_reg - self.droppout = dropout + self.dropout = dropout self.use_bias = use_bias self.early_stop = early_stop self.seed = seed @@ -232,7 +232,7 @@ def _fit_pt(self, train_set, val_set): self.i_biases.reshape(-1, 1), self.use_bias, self.global_mean, - self.droppout, + self.dropout, ) learn(