hbp5181 commited on
Commit
201f12a
·
verified ·
1 Parent(s): a712768

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -17,7 +17,7 @@ tags:
17
  ## ESM2-GBT: Gradient Boosted Trees on ESM2 Embeddings
18
 
19
  # Model Overview
20
- The ESM2-GBT model is a Gradient Boosted Trees (GBT) regressor trained on ESM2 embeddings from Meta’s ESM2 protein language model. It is designed for binding affinity predictive tasks.
21
 
22
 
23
  # Available Pretrianed Models:
@@ -48,7 +48,7 @@ from huggingface_hub import hf_hub_download
48
 
49
  # Download ACE2 RBD model/General model
50
 
51
- model_path = hf_hub_download(repo_id="hbp5181/ESM2-GBT", filename="ACE2_RBD_ESM2-GBT.json")
52
 
53
  Load Model in CatBoost
54
 
@@ -56,7 +56,7 @@ from catboost import CatBoostRegressor
56
 
57
  model = CatBoostRegressor()
58
 
59
- model.load_model(model_path, format="json")
60
 
61
 
62
  # Training Details
 
17
  ## ESM2-GBT: Gradient Boosted Trees on ESM2 Embeddings
18
 
19
  # Model Overview
20
+ The BindPre model is a Gradient Boosted Trees (GBT) regressor trained on ESM2 embeddings from Meta’s ESM2 protein language model. It is designed for binding affinity predictive tasks.
21
 
22
 
23
  # Available Pretrianed Models:
 
48
 
49
  # Download ACE2 RBD model/General model
50
 
51
+ model_path = hf_hub_download(repo_id="hbp5181/BindPre", filename="ACE2_RBD_BindPre.cbm")
52
 
53
  Load Model in CatBoost
54
 
 
56
 
57
  model = CatBoostRegressor()
58
 
59
+ model.load_model(model_path, format="cbm")
60
 
61
 
62
  # Training Details