Beispiel #1
0
Run nested cross-validation experiment on the CCLE IC drug sensitivity dataset, with 
the Gaussian + Gaussian + Volume Prior model.
'''

project_location = "/Users/thomasbrouwer/Documents/Projects/libraries/"
import sys
sys.path.append(project_location)

from BMF_Priors.code.models.bmf_gaussian_gaussian_volumeprior import BMF_Gaussian_Gaussian_VolumePrior
from BMF_Priors.code.cross_validation.nested_matrix_cross_validation import MatrixNestedCrossValidation
from BMF_Priors.data.drug_sensitivity.load_data import load_ccle_ic50_integer


''' Settings BMF model. '''
method = BMF_Gaussian_Gaussian_VolumePrior
R, M = load_ccle_ic50_integer()
hyperparameters = { 'alpha':1., 'beta':1., 'lamb':0.1, 'gamma':10**-10 }
train_config = {
    'iterations' : 150,
    'init' : 'random',
}
predict_config = {
    'burn_in' : 100,
    'thinning' : 1,
}


''' Settings nested cross-validation. '''
K_range = [4,5,6,7]
no_folds = 5
no_threads = 5
Run nested cross-validation experiment on the CCLE IC drug sensitivity dataset, with 
the Gaussian + Exponential + ARD model.
'''

import sys, os
project_location = os.path.dirname(__file__)+"/../../../../"
sys.path.append(project_location)

from BMF_Priors.code.models.bmf_gaussian_exponential_ard import BMF_Gaussian_Exponential_ARD
from BMF_Priors.code.cross_validation.nested_matrix_cross_validation import MatrixNestedCrossValidation
from BMF_Priors.data.drug_sensitivity.load_data import load_ccle_ic50_integer


''' Settings BMF model. '''
method = BMF_Gaussian_Exponential_ARD
R, M = load_ccle_ic50_integer()
hyperparameters = { 'alpha':1., 'beta':1., 'alpha0':1., 'beta0':1. }
train_config = {
    'iterations' : 250,
    'init' : 'random',
}
predict_config = {
    'burn_in' : 200,
    'thinning' : 1,
}


''' Settings nested cross-validation. '''
K_range = [3,4,5,6,7]
no_folds = 5
no_threads = 5