Esempio n. 1
0
        public void TrainBaseModel(int cFeatures)
        {
            double avgRating = getAverageRating(trainUsers); //calculate the average rating of the training set

            IPredictionModel matrixFactorizationModel = new MatrixFactorizationModel(trainUsers, trainItems, validationUsers, validationItems, cFeatures, avgRating);

            predictionEngine.addModel(PredictionMethod.BaseModel, matrixFactorizationModel);
            predictionEngine.Train(PredictionMethod.BaseModel);
        }
Esempio n. 2
0
 public BaseModelMethod(IPredictionModel model)
 {
     this.model = (MatrixFactorizationModel)model;
 }