加入CODE CHINA

· 不限速    · 不限空间    · 不限人数    · 私仓免费

免费加入
readme.md

DeepFM in Keras

Introduction

A simple DeepFM. See details at here.

Environments

  • Keras 2.0.8
  • TensorFlow 1.7

Usage

    python DeepFM.py

Reference

DeepFM: A Factorization-Machine based Neural Network for CTR Prediction paper address

Notes

If you meet errors about Embedding Layer, try fix the compute_mask function.

    def compute_mask(self, inputs, mask=None):
        if not self.mask_zero:
            return None
        else:
            # return K.not_equal(inputs, 0)
            mask = K.repeat(K.not_equal(inputs, 0), self.output_dim)
            mask = tf.transpose(mask, [0,2,1])
            return mask

项目简介

🚀 Github 镜像仓库 🚀

源项目地址

https://github.com/songdark/deepfm_keras

发行版本

当前项目没有发行版本

贡献者 1

S SongDark @SongDark

开发语言

  • Python 100.0 %