models icon indicating copy to clipboard operation
models copied to clipboard

Merlin Models is a collection of deep learning recommender system model reference implementations

Results 205 models issues
Sort by recently updated
recently updated
newest added

I ran into this error when running `prepare_aliccp()` on downloaded Ali-CCP datasets. ``` Traceback (most recent call last): File "/share/suh-scrap/zh338/aliccp/preprocess.py", line 13, in prepare_aliccp(DATA_DIR, convert_train=False, convert_test=True) File "/home/zh338/.conda/envs/merlin-env/lib/python3.10/site-packages/merlin/datasets/ecommerce/aliccp/dataset.py", line 164,...

bug
status/needs-triage

Setup single GPU tests to use RAPIDS GitHub Actions Runner - `gpu-ci` - Runs in `nvcr.io/nvidia/tensorflow:23.06-tf2-py3` - `gpu-ci-cu11` - Runs in `nvidia/cuda:11.8.0-devel-ubuntu22.04`

ci

### Bug description When I try to use the `class_weight` arg in `.fit()` method of `DLRMModel`, I got the following error: ![image](https://github.com/NVIDIA-Merlin/models/assets/46781050/b68218a0-3853-4c34-9136-7170cdb0ffe6) Can someone know why please ? As shown...

bug
status/needs-triage

First I run docker given command ``` docker run --gpus all --rm -it \ -p 8888:8888 -p 8797:8787 -p 8796:8786 --ipc=host \ nvcr.io/nvidia/merlin/merlin-tensorflow-training:latest /bin/bash ``` I want to try first...

status/needs-triage

# 🌟 New model addition ## Model description This model is a slight improvement over the TwoTower architecture as it allows the two towers to interact not just on the...

# ❓ Questions & Help ## Details I want to built ranking model using pre-train embeddings 1) I can train a model using embedding lookups, but the input of the...

status/needs-triage

# ❓ Questions & Help What is the preferred way of generating predictions from a trained `Encoder` from a `TwoTowerModelV2`? There seem to be at least two ways of doing...

status/needs-triage

# ❓ Questions & Help I'm a beginner with Merlin Models. I'm setting up a DLRM model, with 3 types of input features: 1. categorical features 2. continuous features 3....

question
status/needs-triage

### Bug description I am running experiments using sampled softmax: I am using new method (topk_encoder) to evaluate when am using sampled softmax. However, when am not using sampled softmax,...

bug
P1

Ports [lit-llama](https://github.com/Lightning-AI/lit-llama/blob/main/lit_llama/model.py) as a `Block`.