Update README.md
Browse files
README.md
CHANGED
|
@@ -17,7 +17,7 @@ tags:
|
|
| 17 |
## ESM2-GBT: Gradient Boosted Trees on ESM2 Embeddings
|
| 18 |
|
| 19 |
# Model Overview
|
| 20 |
-
The
|
| 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/
|
| 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="
|
| 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
|