AlienChen commited on
Commit
99cc5ad
·
verified ·
1 Parent(s): 2237f88

Update moppit.py

Browse files
Files changed (1) hide show
  1. moppit.py +4 -3
moppit.py CHANGED
@@ -44,12 +44,13 @@ if 'Affinity' in args.objectives:
44
  affinity_predictor = load_affinity_predictor(device)
45
  affinity_model = AffinityModel(affinity_predictor, target_sequence, device)
46
  score_models.append(affinity_model)
47
- if 'Motif' in args.objectives or 'Specificity' in args.objectives:
48
- bindevaluator = load_bindevaluator('./classifier_ckpt/finetuned_BindEvaluator.ckpt', device)
49
- if 'Specificity' in args.objectives:
50
  motif_penalty = True
51
  else:
52
  motif_penalty = False
 
 
53
  motif_model = MotifModel(bindevaluator, target_sequence['input_ids'], motifs, penalty=motif_penalty)
54
  score_models.append(motif_model)
55
 
 
44
  affinity_predictor = load_affinity_predictor(device)
45
  affinity_model = AffinityModel(affinity_predictor, target_sequence, device)
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)
55
  score_models.append(motif_model)
56