Hugging Face's logo

ChatterjeeLab
/
moPPIt

Model card Files
xet
Community
AlienChen commited on
Commit
c20ca8e
·
verified ·
1 Parent(s): 8568186

Update moppit.py

Browse files
Files changed (1) hide show
  1. moppit.py +3 -3
moppit.py CHANGED
@@ -29,12 +29,12 @@ motifs = parse_motifs(motifs).to(device)
29
  print(motifs)
30
 
31
  # Load Models
32
- solver = load_solver('/scratch/pranamlab/tong/checkpoints/MOG-DFM/ckpt/peptide/cnn_epoch200_lr0.0001_embed512_hidden256_loss3.1051.ckpt', vocab_size, device)
33
 
34
- bindevaluator = load_bindevaluator('/scratch/pranamlab/tong/checkpoints/BindEvaluator/model_path/finetuned_BindEvaluator.ckpt', device)
35
  motif_model = MotifModel(bindevaluator, target_sequence, motifs, penalty=True)
36
 
37
- affinity_predictor = load_affinity_predictor('/scratch/pranamlab/tong/checkpoints/MOG-DFM/classifier_ckpt/binding_affinity_unpooled.pt', device)
38
  affinity_model = AffinityModel(affinity_predictor, target_sequence)
39
 
40
  score_models = [motif_model, affinity_model]
 
29
  print(motifs)
30
 
31
  # Load Models
32
+ solver = load_solver('./ckpt/peptide/cnn_epoch200_lr0.0001_embed512_hidden256_loss3.1051.ckpt', vocab_size, device)
33
 
34
+ bindevaluator = load_bindevaluator('./classifier_ckpt/finetuned_BindEvaluator.ckpt', device)
35
  motif_model = MotifModel(bindevaluator, target_sequence, motifs, penalty=True)
36
 
37
+ affinity_predictor = load_affinity_predictor('./classifier_ckpt/binding_affinity_unpooled.pt', device)
38
  affinity_model = AffinityModel(affinity_predictor, target_sequence)
39
 
40
  score_models = [motif_model, affinity_model]