Reset norms when refitting [skip ci]

ankane · Feb 20, 2021 · de2f05e · de2f05e
1 parent ae7a457
commit de2f05e
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/disco/recommender.rb b/lib/disco/recommender.rb
@@ -78,6 +78,9 @@ def fit(train_set, validation_set: nil)
       @user_factors = model.p_factors(format: :numo)
       @item_factors = model.q_factors(format: :numo)
 
+      @user_norms = nil
+      @item_norms = nil
+
       @user_recs_index = nil
       @similar_users_index = nil
       @similar_items_index = nil