Exemplo n.º 1
0
'''
Run nested cross-validation experiment on the CCLE EC drug sensitivity dataset, with 
the Gaussian + L21 model.
'''

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

from BMF_Priors.code.models.bmf_gaussian_l21 import BMF_Gaussian_L21
from BMF_Priors.code.cross_validation.nested_matrix_cross_validation import MatrixNestedCrossValidation
from BMF_Priors.data.drug_sensitivity.load_data import load_ccle_ec50_integer
''' Settings BMF model. '''
method = BMF_Gaussian_L21
R, M = load_ccle_ec50_integer()
hyperparameters = {'alpha': 1., 'beta': 1., 'lamb': 0.1}
train_config = {
    'iterations': 250,
    'init': 'random',
}
predict_config = {
    'burn_in': 200,
    'thinning': 1,
}
''' Settings nested cross-validation. '''
K_range = [1, 2, 3]
no_folds = 5
no_threads = 5
stratify_rows = False
parallel = False
folder_results = './results/gaussian_l21/'
Run nested cross-validation experiment on the CCLE EC drug sensitivity dataset, with 
the All Gaussian model (univariate posterior).
'''

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

from BMF_Priors.code.models.bmf_gaussian_gaussian_univariate import BMF_Gaussian_Gaussian_univariate
from BMF_Priors.code.cross_validation.nested_matrix_cross_validation import MatrixNestedCrossValidation
from BMF_Priors.data.drug_sensitivity.load_data import load_ccle_ec50_integer


''' Settings BMF model. '''
method = BMF_Gaussian_Gaussian_univariate
R, M = load_ccle_ec50_integer()
hyperparameters = { 'alpha':1., 'beta':1., 'lamb':0.1 }
train_config = {
    'iterations' : 250,
    'init' : 'random',
}
predict_config = {
    'burn_in' : 200,
    'thinning' : 1,
}


''' Settings nested cross-validation. '''
K_range = [1,2,3]
no_folds = 5
no_threads = 5