Upload __init__.py with huggingface_hub
Browse files- __init__.py +48 -0
__init__.py
ADDED
|
@@ -0,0 +1,48 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
"""
|
| 2 |
+
Two-Tower Recommendation Module
|
| 3 |
+
|
| 4 |
+
Isengard (User Tower) & Mordor (Wine Tower) for learned wine recommendations.
|
| 5 |
+
"""
|
| 6 |
+
|
| 7 |
+
from .config import (
|
| 8 |
+
EMBEDDING_DIM,
|
| 9 |
+
USER_VECTOR_DIM,
|
| 10 |
+
WINE_VECTOR_DIM,
|
| 11 |
+
HIDDEN_DIM,
|
| 12 |
+
CATEGORICAL_FEATURES,
|
| 13 |
+
CATEGORICAL_VOCAB_SIZES,
|
| 14 |
+
CATEGORICAL_ENCODING_DIM,
|
| 15 |
+
HF_MODEL_ID,
|
| 16 |
+
)
|
| 17 |
+
|
| 18 |
+
from .user_tower import UserTower
|
| 19 |
+
from .wine_tower import WineTower, encode_categorical_features
|
| 20 |
+
from .model import TwoTowerModel, create_training_script
|
| 21 |
+
from .hf_inference_client import (
|
| 22 |
+
TwoTowerHFClient,
|
| 23 |
+
get_hf_client,
|
| 24 |
+
is_two_tower_available,
|
| 25 |
+
)
|
| 26 |
+
|
| 27 |
+
__all__ = [
|
| 28 |
+
# Config
|
| 29 |
+
"EMBEDDING_DIM",
|
| 30 |
+
"USER_VECTOR_DIM",
|
| 31 |
+
"WINE_VECTOR_DIM",
|
| 32 |
+
"HIDDEN_DIM",
|
| 33 |
+
"CATEGORICAL_FEATURES",
|
| 34 |
+
"CATEGORICAL_VOCAB_SIZES",
|
| 35 |
+
"CATEGORICAL_ENCODING_DIM",
|
| 36 |
+
"HF_MODEL_ID",
|
| 37 |
+
# Towers
|
| 38 |
+
"UserTower",
|
| 39 |
+
"WineTower",
|
| 40 |
+
"encode_categorical_features",
|
| 41 |
+
# Model
|
| 42 |
+
"TwoTowerModel",
|
| 43 |
+
"create_training_script",
|
| 44 |
+
# HF Client
|
| 45 |
+
"TwoTowerHFClient",
|
| 46 |
+
"get_hf_client",
|
| 47 |
+
"is_two_tower_available",
|
| 48 |
+
]
|