← all repositories

NVIDIA/DeepRecommender

Deep autoencoder implementation for collaborative filtering-based recommendation systems.

1.7k stars Python Domain Apps
DeepRecommender
Velocity · 7d
+0.5
★ / day
Trend
steady
star history

This repository implements deep autoencoders for collaborative filtering, as described in a research paper. The model is built with PyTorch and designed to run on GPU with CUDA support. It includes mixed precision training capabilities using NVIDIA Tensor Cores on Volta GPUs. The project demonstrates training on the Netflix Prize dataset for recommendation tasks.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.