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