Downloads · 30 days
10
45% of all-time downloads
zjkarina/omniRecsysLLM_idmodality
omniRecsysLLM_idmodality is a machine learning model from zjkarina. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
Recommendation model based on ID modality for Amazon Fashion.
Downloads · 30 days
10
45% of all-time downloads
All-time downloads
22
Public
Repo size
37.2 GB
Likes
0
Public
Click a slice to open those files.
.pt37.2 GB · 100%
From the Hugging Face model README
Recommendation model based on ID modality for Amazon Fashion.
This model uses item ID embeddings to generate recommendations based on users’ purchase history.
from any2any_trainer.models.recommendation import RecommendationModel
# Load model
model = RecommendationModel.from_pretrained("zjkarina/omniRecsysLLM_idmodality")
# Generate recommendations
recommendations = model.predict_next_item(
text="The user bought jeans and a t-shirt",
id_ids=[12345, 67890], # Item IDs from purchase history
top_k=5
)
The model was trained on the Amazon Fashion 2023 dataset using the ID modality.