Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -110,7 +110,7 @@ import numpy as np
|
|
| 110 |
|
| 111 |
# Load best model (7T-21T, K=1, Euclidean)
|
| 112 |
model = AutoModel.from_pretrained(
|
| 113 |
-
"
|
| 114 |
trust_remote_code=True
|
| 115 |
)
|
| 116 |
|
|
@@ -128,7 +128,7 @@ print(predictions)
|
|
| 128 |
```python
|
| 129 |
# Load 21T model with K=1 and Euclidean distance
|
| 130 |
model = AutoModel.from_pretrained(
|
| 131 |
-
"
|
| 132 |
data_source="21T",
|
| 133 |
k_neighbors=1,
|
| 134 |
metric="euclidean",
|
|
@@ -137,7 +137,7 @@ model = AutoModel.from_pretrained(
|
|
| 137 |
|
| 138 |
# Load 7T-21T ensemble (automatically loads 2 sub-models)
|
| 139 |
model = AutoModel.from_pretrained(
|
| 140 |
-
"
|
| 141 |
data_source="7T-21T",
|
| 142 |
k_neighbors=1,
|
| 143 |
metric="euclidean",
|
|
|
|
| 110 |
|
| 111 |
# Load best model (7T-21T, K=1, Euclidean)
|
| 112 |
model = AutoModel.from_pretrained(
|
| 113 |
+
"SaeedLab/dom-formula-assignment-using-knn",
|
| 114 |
trust_remote_code=True
|
| 115 |
)
|
| 116 |
|
|
|
|
| 128 |
```python
|
| 129 |
# Load 21T model with K=1 and Euclidean distance
|
| 130 |
model = AutoModel.from_pretrained(
|
| 131 |
+
"SaeedLab/dom-formula-assignment-using-knn",
|
| 132 |
data_source="21T",
|
| 133 |
k_neighbors=1,
|
| 134 |
metric="euclidean",
|
|
|
|
| 137 |
|
| 138 |
# Load 7T-21T ensemble (automatically loads 2 sub-models)
|
| 139 |
model = AutoModel.from_pretrained(
|
| 140 |
+
"SaeedLab/dom-formula-assignment-using-knn",
|
| 141 |
data_source="7T-21T",
|
| 142 |
k_neighbors=1,
|
| 143 |
metric="euclidean",
|