tabito123/gohan-product-recommendation-v2

Gohan Product Recommendation Model

This model provides product recommendations for Gohan (rice) products using FT-Transformer architecture.

Model Information

• Architecture: FT-Transformer
• Training Epochs: 30
• Validation Performance: 0.7736
• Input Features: Categorical features
• Output: Product recommendations

Usage

from inference import load_model, predict

# Load the model
model = load_model()

# Make predictions
predictions = predict(model, input_data)

Files

• epoch_030_p30_0.7736.pt: Trained PyTorch model (Git LFS)
• gohan_product_master_data.csv: Product master data
• encoders/: JSON-encoded feature encoders
• configs/config.json: Model configuration

License

Apache-2.0