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

saving work

Browse source
This commit is contained in:
Tanushree Tunstall 2022-07-29 00:12:43 +01:00
parent 1695e90b42
commit e55906d2c7
3 changed files with 11 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -269,7 +269,9 @@ def split_tts(ml_input_data
#k_sm = 5 # default
k_sm = k_smote
sm_nc = SMOTENC(categorical_features=categorical_colind, k_neighbors = k_sm, **rs, **njobs)
sm_nc = SMOTENC(categorical_features=categorical_colind
, k_neighbors = k_sm
, **rs, **njobs)
X_smnc, y_smnc = sm_nc.fit_resample(X, y)
print('\nSMOTE_NC OverSampling\n', Counter(y_smnc))
print(X_smnc.shape)