HAOChuzhan / DeepFM-MindSpore Public

Notifications You must be signed in to change notification settings
Fork 0
Star 0

Here is a reproduction of DeepFM model in the MindSpore framework, referring to the model_zoo in MindSpore.

0 stars 0 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 5 Commits
src		src
README.md		README.md
auc.log		auc.log
eval.py		eval.py
loss.log		loss.log
requirements.txt		requirements.txt
train.py		train.py

Repository files navigation

MindSpore-DeepFM

这里是对于MinsSpore框架中部分模型的复现，参考其中model_zoo相关模型。

deepFM模型

数据预处理输入： python src/preprocess_data.py
模型训练输入： python train.py
模型评估输入： python eval.py # 需要修改对应--checkpoint_path中的最新ckpt文件位置

About

Here is a reproduction of DeepFM model in the MindSpore framework, referring to the model_zoo in MindSpore.

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%