Skip to content
Project categories

Machine Learning · GitHub

Train, evaluate, and run machine learning models.

Useful GitHub projects

8,052 projects

yolor

WongKinYiu

2,010stars
yolor preview

implementation of paper - You Only Learn One Representation: Unified Network for Multiple Tasks (https://arxiv.org/abs/2105.04206)

ml-by-hand

workofart

110stars
ml-by-hand preview

A deep learning library built from scratch with complex neural networks examples built on top for learning purposes.

gangealing

wpeebles

1,014stars
gangealing preview

Official PyTorch Implementation of "GAN-Supervised Dense Visual Alignment" (CVPR 2022 Oral, Best Paper Finalist)

Machine LearningPythonBSD-2-ClauseGitHub
559stars

Python re-implementation of the (constrained) spectral clustering algorithms used in Google's speaker diarization papers.

Machine LearningPythonApache-2.0GitHub
368stars
demo-chinese-text-binary-classification-with-bert preview

因为我当时正在用 fast.ai 的 ULMfit 做自然语言分类任务(还专门写了《如何用 Python 和深度迁移学习做文本分类?》一文分享给你)。ULMfit 和 BERT 都属于预训练语言模型(Pre-trained Language Modeling),具有很多的相似性。

Machine LearningJupyter NotebookGitHub

PSGAN

wtjiang98

781stars
PSGAN preview

PyTorch code for "PSGAN: Pose and Expression Robust Spatial-Aware GAN for Customizable Makeup Transfer" (CVPR 2020 Oral)

NeuRec

wubinzzu

1,060stars
NeuRec preview

Next RecSys Library

  • Cross platform - run on any operating system with the available Docker images;
  • State-of-the-art - 33 neural recommender models available out of the box;
806stars
Doraemon preview

A powerful baseline for image classification, face recognition and image retrieval with Pytorch

  • Training engine: Unified training flow for classification and embedding-based recognition tasks.
  • Optimization algorithms: SGD, Adam, SAM, layer-specific learning rates, and related training strategies.