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

added random state to split in function

Browse source
This commit is contained in:
Tanushree Tunstall 2022-07-05 14:15:43 +01:00
parent e5f882841e
commit 53c229f480
3 changed files with 24 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

7
scripts/ml/ml_functions/SplitTTS.py
View file

@ -40,7 +40,7 @@ import argparse
import re
homedir = os.path.expanduser("~")
#%% GLOBALS
rs = {'random_state': 42}
#rs = {'random_state': 42}
njobs = {'n_jobs': os.cpu_count() } # the number of jobs should equal the number of CPU cores
#%% Define split_tts function #################################################
@ -51,7 +51,10 @@ def split_tts(ml_input_data
, dst_colname = 'dst'# determine how to subset the actual vs reverse data
, target_colname = 'dst_mode'
, include_gene_name = True
, k_smote = 5):
, k_smote = 5
, random_state = 42):
rs = {'random_state': random_state}
outDict = {}