DIA
/
mody_2024
2
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix

main
ifiguero 2025-10-14 00:15:40 -03:00
parent 78e36c77a5
commit 77815ce0d3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -33,7 +33,7 @@ seeds = [231964, 48928, 132268, 113986, 574626, 130068, 226585, 446306, 535997,
# * Model trained on data with no missing values, and impute MICE and KNN
# * Different random train and test splits, for given test_size ratio
# A folder is created with the label name with all the state and run data
mody2 = BinaryTuner(dms2, 'MODY2_label', seeds=seed, test_size=0.2)
mody2 = BinaryTuner(dms2, 'MODY2_label', seeds=seeds, test_size=0.2)
# CheckPoint during search all trained models are saved inside the label folder
# including the scalers required for inference and testing