Surprise

A specialized Python scikit for building and analyzing recommender systems that handle explicit rating data.

Surprise (Simple Python Recommendation System Engine) provides a robust framework for developers to build custom recommendation engines using built-in prediction algorithms like SVD, PMF, and BaselineOnly. It streamlines the research process by offering tools for cross-validation, hyperparameter tuning with GridSearchCV, and performance metrics including RMSE and MAE. By focusing on explicit data structures, Surprise allows for seamless integration with NumPy arrays and Pandas DataFrames to scale collaborative filtering models efficiently.

https://surpriselib.com