tanu/LSHTM_analysis
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

minor changes to run combined_model

Browse source
This commit is contained in:
Tanushree Tunstall 2022-07-10 13:23:13 +01:00
parent 057c98c2f1
commit 350be30f19
3 changed files with 13 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

17
scripts/ml/combined_model/cm_logo_skf.py
View file

@ -172,26 +172,25 @@ def CMLogoSkf(cm_input_df
cm_bts_y = cm_test_df.loc[:, target_var]
print('\nTEST data dim:' , cm_bts_X.shape
, '\nTEST Target dim:', cm_bts_y.shape)
, '\nTEST Target dim:' , cm_bts_y.shape)
print("Running Multiple models on LOGO with SKF")
#%%:Running Multiple models on LOGO with SKF
cD3_v2 = MultModelsCl_logo_skf(input_df = cm_X
, target = cm_y
#, group = 'none'
, sel_cv = skf_cv
, blind_test_df = cm_bts_X
, blind_test_target = cm_bts_y
, tts_split_type = tts_split_type
, tts_split_type = tts_split_type
, resampling_type = 'none' # default
#, group = 'none'
, add_cm = True
, add_yn = True
, var_type = 'mixed'
, scale_numeric = ['min_max']
, run_blind_test = True
, blind_test_df = cm_bts_X
, blind_test_target = cm_bts_y
, return_formatted_output = True
, random_state = 42
, n_jobs = os.cpu_count() # the number of jobs should equal the number of CPU cores