Joblib
PeptiVerse / training_classifiers /hemolysis /cnn_smiles /optimization_summary.txt
ynuozhang
update code
baf3373
========================================================================
MODEL: cnn
Best Optuna F1 (objective): 0.5290
Best Optuna AUC (val, recorded): 0.7477
Best Optuna threshold (val): 0.4518
Refit best AUC (val): 0.7851
Refit best F1@thr (val): 0.5366 at thr=0.5298
Best params:
{
"lr": 0.0002237456677696451,
"weight_decay": 0.0005722918417016266,
"dropout": 0.2697397384794115,
"batch_size": 16,
"channels": 512,
"kernel": 3,
"layers": 4
}
Saved model: /vast/projects/pranam/lab/yz927/projects/Classifier_Weight/training_classifiers/hemolysis/cnn_smiles/best_model.pt
========================================================================