Sonja Topf commited on
Commit
5567bdb
·
1 Parent(s): 103d273

try out 0.5 default value

Browse files
Files changed (1) hide show
  1. predict.py +3 -3
predict.py CHANGED
@@ -71,9 +71,9 @@ def predict(smiles_list):
71
  predictions[original_smi] = pred_dict
72
 
73
  # Add placeholder predictions for invalid SMILES
74
- # for smi, is_valid in zip(smiles_list, valid_mask):
75
- # if not is_valid:
76
- # predictions[smi] = {t: 0.0 for t in TARGET_NAMES}
77
 
78
 
79
  return predictions
 
71
  predictions[original_smi] = pred_dict
72
 
73
  # Add placeholder predictions for invalid SMILES
74
+ for smi, is_valid in zip(smiles_list, valid_mask):
75
+ if not is_valid:
76
+ predictions[smi] = {t: 0.5 for t in TARGET_NAMES}
77
 
78
 
79
  return predictions