models
models copied to clipboard
Merlin Models is a collection of deep learning recommender system model reference implementations
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,...
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`
### Bug description When I try to use the `class_weight` arg in `.fit()` method of `DLRMModel`, I got the following error:  Can someone know why please ? As shown...
[DOC] Merlin Docker AttributeError: module 'merlin.models.tf' has no attribute 'OutputBlock' Error
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...
# 🌟 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...
# ❓ 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...
# ❓ 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....
### 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,...
Ports [lit-llama](https://github.com/Lightning-AI/lit-llama/blob/main/lit_llama/model.py) as a `Block`.