AlienChen commited on
Commit
65a78ff
·
verified ·
1 Parent(s): 99cc5ad

Update moppit.py

Browse files
Files changed (1) hide show
  1. moppit.py +3 -3
moppit.py CHANGED
@@ -46,9 +46,9 @@ if 'Affinity' in args.objectives:
46
  score_models.append(affinity_model)
47
 
48
  if 'Specificity' in args.objectives:
49
- motif_penalty = True
50
- else:
51
- motif_penalty = False
52
  if 'Motif' in args.objectives or 'Specificity' in args.objectives:
53
  bindevaluator = load_bindevaluator('./classifier_ckpt/finetuned_BindEvaluator.ckpt', device)
54
  motif_model = MotifModel(bindevaluator, target_sequence['input_ids'], motifs, penalty=motif_penalty)
 
46
  score_models.append(affinity_model)
47
 
48
  if 'Specificity' in args.objectives:
49
+ motif_penalty = True
50
+ else:
51
+ motif_penalty = False
52
  if 'Motif' in args.objectives or 'Specificity' in args.objectives:
53
  bindevaluator = load_bindevaluator('./classifier_ckpt/finetuned_BindEvaluator.ckpt', device)
54
  motif_model = MotifModel(bindevaluator, target_sequence['input_ids'], motifs, penalty=motif_penalty)