ChatterjeeLab
/

moPPIt

Model card Files Files and versions

AlienChen commited on 12 days ago

Commit

2c8e2d0

·

verified ·

1 Parent(s): c14a243

Update moppit.py

Files changed (1) hide show

moppit.py +1 -1

moppit.py CHANGED Viewed

@@ -41,7 +41,7 @@ if 'Half-Life' in args.objectives:
     halflife_model = HalfLifeModel(device=device)
     score_models.append(halflife_model)
 if 'Affinity' in args.objectives:
-    affinity_predictor = load_affinity_predictor('../classifier_ckpt/wt_affinity.pt', device)
     affinity_model = AffinityModel(affinity_predictor, target_sequence, device)
     score_models.append(affinity_model)
 if 'Motif' in args.objectives or 'Specificity' in args.objectives:

     halflife_model = HalfLifeModel(device=device)
     score_models.append(halflife_model)
 if 'Affinity' in args.objectives:
+    affinity_predictor = load_affinity_predictor('./classifier_ckpt/wt_affinity.pt', device)
     affinity_model = AffinityModel(affinity_predictor, target_sequence, device)
     score_models.append(affinity_model)
 if 'Motif' in args.objectives or 'Specificity' in args.objectives: