Thresholding and updated the script
To predict, use the below command
python predict.py -test_path ./test_data/ -test_outpath ./out/phyto_out_2 / -main_param_path ./out/phyto_out_2/ -model_path ./out/phyto_out_2/trained_models/Init_0/ ./out/phyto_out_2/trained_models/Init_1/ ./out/phyto_out_2/trained_models/Init_2/ ./out/phyto_out_2/trained_models/Init_3/ ./out/phyto_out_2/trained_models/Init_4/ -ensemble 2 -finetuned 2 -threshold 0.0