Update moppit.py
Browse files
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 |
-
|
| 50 |
-
|
| 51 |
-
|
| 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)
|