Update README.md
Browse files
README.md
CHANGED
|
@@ -8,7 +8,8 @@ tags:
|
|
| 8 |
- bilinear-networks
|
| 9 |
metrics:
|
| 10 |
- loss
|
| 11 |
-
pipeline_tag:
|
|
|
|
| 12 |
---
|
| 13 |
|
| 14 |
# SBL-NET (Scalar Bilinear Linear Network)
|
|
@@ -49,7 +50,7 @@ import torch as t
|
|
| 49 |
from huggingface_hub import hf_hub_download
|
| 50 |
|
| 51 |
# 1. Download the architecture and weights files (replace YOUR_USERNAME)
|
| 52 |
-
REPO_ID = "
|
| 53 |
hf_hub_download(repo_id=REPO_ID, filename="model.py", local_dir=".")
|
| 54 |
hf_hub_download(repo_id=REPO_ID, filename="model_weights_hybrid.pth", local_dir=".")
|
| 55 |
|
|
|
|
| 8 |
- bilinear-networks
|
| 9 |
metrics:
|
| 10 |
- loss
|
| 11 |
+
pipeline_tag: tabular-regression
|
| 12 |
+
|
| 13 |
---
|
| 14 |
|
| 15 |
# SBL-NET (Scalar Bilinear Linear Network)
|
|
|
|
| 50 |
from huggingface_hub import hf_hub_download
|
| 51 |
|
| 52 |
# 1. Download the architecture and weights files (replace YOUR_USERNAME)
|
| 53 |
+
REPO_ID = "ALEX_FLR/sbl-net"
|
| 54 |
hf_hub_download(repo_id=REPO_ID, filename="model.py", local_dir=".")
|
| 55 |
hf_hub_download(repo_id=REPO_ID, filename="model_weights_hybrid.pth", local_dir=".")
|
| 56 |
|