| """ | |
| 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", | |
| ] | |