'''
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,
