PreferredAI/cornac

A Comparative Framework for Multimodal Recommender Systems

解决的问题

Cornac 是一个旨在简化多模态推荐系统开发、评估和部署的比较框架。它解决了将辅助数据(如物品描述、图像和社交网络)整合到推荐模型中的困难,并提供了一种标准化的方法,用于横向比较不同算法。

工作原理

Cornac 提供统一的接口用于加载数据集、划分数据用于评估,以及实现各种推荐模型。它与 PyTorch 和 TensorFlow 等流行机器学习库集成。框架内置 Experiment 类,允许用户同时在相同指标和数据集上运行多个模型,以确定表现最佳的算法。

适用人群

适用于需要快速实验、实现新模型并确保结果可复现的研究人员和开发者。

主要亮点

  • 多模态支持:特别优化用于利用文本和图像等辅助数据的模型。
  • 丰富的模型库:包含从协同过滤、下一物品预测到生成式检索和可解释推荐的广泛实现模型。
  • 部署工具:提供基于 Flask 的内置服务应用,可快速部署模型,并与近似最近邻(ANN)搜索框架(如 Faiss、HNSWLib)集成,实现高效检索。
  • 评估框架:被 ACM RecSys 2023 推荐用于评估和可复现性。

相关

  • 项目
  • 项目
  • 项目
  • 项目
  • 项目