Update moppit.py
Browse files
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 |
-
|
| 48 |
-
|
| 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 |
|