swirl commited on
Commit
a272c05
·
verified ·
1 Parent(s): 88e6227

Upload __init__.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. __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
+ ]