SkillAgentSearch skills...

Mlgb

MLGB is a library that includes many models of CTR Prediction & Recommender System by TensorFlow & PyTorch. 「妙计包」是一个包含50+点击率预估和推荐系统深度模型的、通过TensorFlow和PyTorch撰写的库。

Install / Use

/learn @UlionTse/Mlgb

README

<p align="center"> <img src="https://github.com/UlionTse/mlgb/blob/main/docs/mlgb_logo.png" width="200"/> </p> <p align="center"> <a href="https://pypi.org/project/mlgb"><img alt="PyPI - Version" src="https://img.shields.io/pypi/v/mlgb.svg?color=blue"></a> <a href="https://anaconda.org/conda-forge/mlgb"><img alt="Conda - Version" src="https://img.shields.io/conda/vn/conda-forge/mlgb.svg?color=blue"></a> <a href="https://pypi.org/project/mlgb"><img alt="PyPI - License" src="https://img.shields.io/pypi/l/mlgb.svg?color=brightgreen"></a> <a href="https://pypi.org/project/mlgb"><img alt="PyPI - Python" src="https://img.shields.io/pypi/pyversions/mlgb.svg?color=blue"></a> <a href="https://pypi.org/project/mlgb"><img alt="PyPI - Status" src="https://img.shields.io/pypi/status/mlgb.svg?color=brightgreen"></a> <a href="https://pypi.org/project/mlgb"><img alt="PyPI - Wheel" src="https://img.shields.io/badge/wheel-yes-brightgreen.svg"></a> <a href="https://pypi.org/project/mlgb"><img alt="PyPI - Downloads" src="https://static.pepy.tech/personalized-badge/mlgb?period=total&units=international_system&left_text=downloads&left_color=grey&right_color=blue"></a> <a href="https://pypi.org/project/mlgb"><img alt="PyPI - TensorFlow" src="https://img.shields.io/badge/TensorFlow-2.10+-yellow.svg"></a> <a href="https://pypi.org/project/mlgb"><img alt="PyPI - PyTorch" src="https://img.shields.io/badge/PyTorch-2.1+-tomato.svg"></a> </p>

MLGB means Machine Learning of the Great Boss, and is called 「妙计包」.
MLGB is a library that includes many models of CTR Prediction & Recommender System by TensorFlow & PyTorch.

Advantages

  • Easy! Use mlgb.get_model(model_name, **kwargs) to get a complex model.
  • Fast! Better performance through better code.
  • Enjoyable! 50+ ranking & matching models to use, 2 languages(TensorFlow & PyTorch) to deploy.
  • PaperWithCode

Supported Models

| ID | Model Name | Paper Link | Paper Team | Paper Year | | --- | ------------- | -------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------- | ---------------------------------------------------------------------------- | ---------- | | <tr><th colspan=5 align="center">:open_file_folder: Ranking-Model::Normal :point_down:</th></tr> | | 1 | LR | Predicting Clicks: Estimating the Click-Through Rate for New Ads | Microsoft | 2007 | | 2 | PLM/MLR | Learning Piece-wise Linear Models from Large Scale Data for Ad Click Prediction | Alibaba | 2017 | | 3 | MLP/DNN | Neural Networks for Pattern Recognition | Christopher M. Bishop(Microsoft, 1997-Present), Foreword by Geoffrey Hinton. | 1995 | | 4 | DLRM | Deep Learning Recommendation Model for Personalization and Recommendation Systems | Facebook(Meta) | 2019 | | 5 | MaskNet | MaskNet: Introducing Feature-Wise Multiplication to CTR Ranking Models by Instance-Guided Mask | Weibo(Sina) | 2021 | | | | | | | | 6 | DCM/DeepCross | Deep Crossing: Web-Scale Modeling without Manually Crafted Combinatorial Features | Microsoft | 2016 | | 7 | DCN | DCN V2: Improved Deep & Cross Network and Practical Lessons for Web-scale Learning to Rank Systems, v1 | Google(Alphabet) | 2017, 2020 | | 8 | EDCN | Enhancing Explicit and Implicit Feature Interactions via Information Sharing for Parallel Deep CTR Models | Huawei | 2021 | | | | | | | | 9 | FM | Factorization Machines | Steffen Rendle(Google, 2013-Present) | 2010 | | 10 | FFM | Field-aware Factorization Machines for CTR Prediction | NTU | 2016 | | 11 | HOFM | Higher-Order Factorization Machines | NTT | 2016 | | 12 | FwFM | Field-weighted Factorization Machines for Click-Through Rate Prediction in Display Advertising | Junwei Pan(Yahoo), etc. | 2018, 2020 | | 13 | FmFM | FM^2: Field-matrixed Factorization Machines for Recommender Systems | Yahoo | 2021 | | 14 | FEFM | FIELD-EMBEDDED FACTORIZATION MACHINES FOR CLICK-THROUGH RATE PREDICTION | Harshit Pande(Adobe) | 2020, 2021 | | 15 | AFM | Attentional Factorization Machines: Learning the Weight of Feature Interactions via Attention Networks | ZJU&NUS(Jun Xiao(ZJU), Xiangnan He(NUS), etc.) | 2017 | | 16 | LFM | Learning Feature Interactions with Lorentzian Factorization Machine | EBay | 2019 | | 17 | IFM | An Input-aware Factorization Machine for Sparse Prediction | THU | 2019 | | 18 | DIFM | A Dual Input-aware Factorization Machine for CTR Prediction | THU | 2020 | | | | | | | | 19 | FNN | Deep Learning over Multi-field Categorical Data – A Case Study on User Response Prediction | UCL(Weinan Zhang(UCL, SJTU), etc.) | 2016 | | 20 | PNN | Product-based Neural Networks for User Response | SJTU&UCL(Yanru Qu(SJTU), Weinan Zhang(SJTU, UCL), etc.) | 2016 | | 21 | PIN | [Product-based Neural Networks for User Response Prediction over Multi-field Categorical

View on GitHub
GitHub Stars1.0k
CategoryEducation
Updated1d ago
Forks95

Languages

Python

Security Score

100/100

Audited on Apr 1, 2026

No findings