Shakiran's picture
Upload 8 files
ab3a223 verified
from nutrition_ensemble import NutritionEnsemble
# Initialize model
ensemble = NutritionEnsemble(".")
# Example 1: Basic recommendations
print("=== Basic Recommendations ===")
recs = ensemble.recommend(user_id=0, top_k=5)
print(recs)
# Example 2: With exclusions
print("\n=== With Exclusions ===")
recs = ensemble.recommend(user_id=0, top_k=5, exclude_foods=[1, 2, 3])
print(recs)