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); }
public BaseModelMethod(IPredictionModel model) { this.model = (MatrixFactorizationModel)model; }