← all repositories
shenweichen/DeepCTR

Thirty CTR papers, one .fit() call

Because nobody should reimplement DeepFM from scratch just to run a baseline.

8.1k stars Python ML FrameworksDomain Apps
DeepCTR
Not currently ranked — collecting fresh signals.
star history

What it does DeepCTR packages roughly 30 published click-through-rate architectures into a single TensorFlow library. You get tf.keras-style objects for models ranging from Wide & Deep and DeepFM to attention-based and multi-task variants like DIN, DIEN, ESMM, and PLE, all callable via model.fit() and model.predict(). It also exposes the underlying feature-crossing and attention layers, so you can assemble custom hybrids without copying code from scattered paper repos.

The interesting bit The project treats CTR research like a parts catalog rather than a collection of standalone demos. Swapping from DeepFM to xDeepFM is an import change, not a repo migration, and the modular layers let you mix attention mechanisms, explicit feature crosses, and multi-task heads in custom combinations.

Key highlights

  • Implements roughly 30 published CTR models (2015–2021) in one codebase
  • Offers both tf.keras.Model and TensorFlow Estimator interfaces
  • Compatible with TensorFlow 1.15 and 2.x
  • Exposes reusable core layers for custom model assembly
  • Contributors include engineers from Alibaba and Amazon

Caveats

  • Does not bundle TensorFlow; compatibility notes warn about NumPy and h5py version conflicts on Python ≥3.9
  • The README is essentially a model checklist with paper links; guidance on which architecture fits which dataset shape or training budget is absent
  • PyTorch users need the separate DeepCTR-Torch project

Verdict Worth a look if you need to benchmark CTR baselines or prototype new architectures from existing blocks. Skip it if you want a managed training pipeline, performance comparisons, or a PyTorch stack.

Frequently asked

What is shenweichen/DeepCTR?
Because nobody should reimplement DeepFM from scratch just to run a baseline.
Is DeepCTR open source?
Yes — shenweichen/DeepCTR is open source, released under the Apache-2.0 license.
What language is DeepCTR written in?
shenweichen/DeepCTR is primarily written in Python.
How popular is DeepCTR?
shenweichen/DeepCTR has 8.1k stars on GitHub.
Where can I find DeepCTR?
shenweichen/DeepCTR is on GitHub at https://github.com/shenweichen/DeepCTR.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.