Open-source alternatives to Spotlight

This project is a PyTorch implementation of the Faster R-CNN architecture for object detection. It provides a framework for identifying multiple object classes and their corresponding bounding boxes within images using a deep learning system. The implementation includes a training pipeline for optimizing models on custom datasets and a utility for converting pretrained weights from external formats into a compatible structure for model initialization. The system covers a two-stage detection pipeline comprising a region proposal network and an ROI pooling layer. It incorporates multi-task los

Super simple fit method for PyTorch Modules

Torch7 -> CoreML

A framework for large scale recommendation algorithms.

Fast Python Collaborative Filtering for Implicit Feedback Datasets

This project is a pretrained model library for PyTorch, providing a collection of convolutional neural network architectures and weights. It serves as a computer vision model zoo for image classification and feature extraction, offering a framework for transfer learning where pretrained networks are adapted for custom image recognition tasks. The library focuses on transforming images into high-level numerical representations and calculating class probability scores. It includes utilities for downloading and initializing standard architectures such as ResNet, Inception, and Xception. Capabil

This project is a machine learning experiment tracker and event file generator that enables the recording of scalars, images, and histograms to monitor model performance. It functions as an integration bridge that allows training metrics from PyTorch to be logged into files compatible with the TensorBoard dashboard. The system includes a remote log synchronizer designed to stream experiment data to cloud services. This allows for the remote management and analysis of training results and the comparison of datasets across different training runs. The utility covers a broad range of monitoring

LightFM is a Python recommendation library and machine learning framework designed to predict user preferences. It implements a hybrid recommendation engine that combines collaborative filtering with content filtering by integrating user-item interaction data with descriptive metadata. The system utilizes hybrid matrix factorization to learn latent representations of users and items. It is specifically designed to handle implicit feedback, utilizing specialized loss functions such as Weighted Approximate Rank Pairwise and Bayesian Personalized Ranking to optimize item preferences for datasets

Train AI models efficiently on medical images using any framework