''' Run nested cross-validation experiment on the MovieLens 1M dataset, with the All Gaussian model (multivariate posterior). ''' project_location = "/home/tab43/Documents/Projects/libraries/" # "/Users/thomasbrouwer/Documents/Projects/libraries/" import sys sys.path.append(project_location) from BMF_Priors.code.models.bmf_gaussian_gaussian import BMF_Gaussian_Gaussian from BMF_Priors.code.cross_validation.nested_matrix_cross_validation import MatrixNestedCrossValidation from BMF_Priors.data.movielens.load_data import load_processed_movielens_1M ''' Settings BMF model. ''' method = BMF_Gaussian_Gaussian R, M = load_processed_movielens_1M() hyperparameters = {'alpha': 1., 'beta': 1., 'lamb': 0.1} train_config = { 'iterations': 120, 'init': 'random', } predict_config = { 'burn_in': 100, 'thinning': 1, } ''' Settings nested cross-validation. ''' K_range = [3, 4, 5, 6, 7] no_folds = 5 no_threads = 5 parallel = False folder_results = './results/gaussian_gaussian/' output_file = folder_results + 'results.txt'
Run cross-validation experiment on the MovieLens 1M dataset, with the row-average baseline. ''' import sys, os project_location = os.path.dirname(__file__)+"/../../../../" sys.path.append(project_location) from BMF_Priors.code.models.baseline_average_row import RowAverage from BMF_Priors.code.cross_validation.matrix_single_cross_validation import MatrixSingleCrossValidation from BMF_Priors.data.movielens.load_data import load_processed_movielens_1M ''' Settings BMF model. ''' method = RowAverage R, M = load_processed_movielens_1M() hyperparameters = {} train_config = { 'iterations' : 0, 'init' : '' } predict_config = { 'burn_in': 0, 'thinning': 0 } parameters = {'K':0, 'hyperparameters':hyperparameters} ''' Settings nested cross-validation. ''' no_folds = 5 folder_results = './results/baseline_average_row/' output_file = folder_results+'results.txt' ''' Run the cross-validation framework. ''' crossval = MatrixSingleCrossValidation( method=method,