orange3-recommendation
orange3-recommendation copied to clipboard
SVDPlusPlusLearner traninng diverge error
Recommendation version
latest
Orange version
3.27.1
Expected behavior
(Print on screen) [0.98389692 1.01224489 1.0078577 ]
Steps to reproduce the behavior
import Orange from orangecontrib.recommendation import SVDPlusPlusLearner, BRISMFLearner
#Load data and train the model data = Orange.data.Table('movie_lens/epinions_train.tab') learner = SVDPlusPlusLearner(num_factors=15, num_iter=25, learning_rate=0.07, lmbda=0.1) recommender = learner(data)
#Make predictions prediction = recommender(data[:3]) print(prediction)
Actual behavior
RuntimeError: Training diverged and returned NaN.
Additional info (worksheets, data, screenshots, ...)
data are from the dataset folder of the recommendation package; placed at a subfolder "movie_lens" in the path of the script; actually any of the datafile causes the same error. using BRIMSFLearner works. https://github.com/biolab/orange3-recommendation/blob/master/orangecontrib/recommendation/datasets/epinions_train.tab