tabito123 commited on
Commit
afe4306
·
verified ·
1 Parent(s): 1147a5d

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +44 -0
README.md ADDED
@@ -0,0 +1,44 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ language:
4
+ - ja
5
+ library_name: pytorch
6
+ tags:
7
+ - recommendation
8
+ - tabular
9
+ - ft-transformer
10
+ model_name: gohan-product-recommendation
11
+ repo_type: model
12
+ ---
13
+
14
+ # Gohan Product Recommendation Model
15
+
16
+ This model provides product recommendations for Gohan (rice) products using FT-Transformer architecture.
17
+
18
+ ## Model Information
19
+ - **Architecture**: FT-Transformer
20
+ - **Training Epochs**: 30
21
+ - **Validation Performance**: 0.7736
22
+ - **Input Features**: Categorical features
23
+ - **Output**: Product recommendations
24
+
25
+ ## Usage
26
+
27
+ ```python
28
+ from inference import load_model, predict
29
+
30
+ # Load the model
31
+ model = load_model()
32
+
33
+ # Make predictions
34
+ predictions = predict(model, input_data)
35
+ ```
36
+
37
+ ## Files
38
+ - `epoch_030_p30_0.7736.pt`: Trained PyTorch model (Git LFS)
39
+ - `gohan_product_master_data.csv`: Product master data
40
+ - `encoders/`: JSON-encoded feature encoders
41
+ - `configs/config.json`: Model configuration
42
+
43
+ ## License
44
+ Apache-2.0