← all projects
Machine Learning · April 2024 · 1 min read

Collaborative Filtering Movie Recommender

SVD-based matrix factorization system with 0.89 RMSE on MovieLens dataset. Handles sparse ratings, cold start problems, and scales to 100k+ users through optimized gradient descent.

PythonNumPyMatrix FactorizationCollaborative Filtering

SVD matrix factorization achieving 0.89 RMSE on MovieLens 100K dataset. Handles 94% sparsity through stochastic gradient descent with bias modeling and regularization.

algorithmSVD matrix factorization with bias terms scalability100k users, 1.7k movies, 10M+ potential ratings performance0.89 RMSE, beats all baseline methods

Training Dynamics

SGD Optimization: Loss converges rapidly in 20 epochs. Optimal learning rate α=0.01, regularization λ=0.1 prevents overfitting to sparse patterns.

User Behavior Patterns: Rating distribution skews positive. Cold start performance improves rapidly with just 5 initial ratings.

Latent Factor Analysis: K=50 factors optimal — fewer underfit complex preferences, more overfit to training sparsity. Training time scales linearly with dimensionality.

#recommender-systems#matrix-factorization#svd#optimization

Related projects