为什么这个项目会依赖tensorflow？

shenweichen / DeepCTR-Torch

【PyTorch】Easy-to-use,Modular and Extendible package of deep-learning based CTR models.

Apache License 2.0

2.95k stars 696 forks source link

Open ralgond opened 1 year ago

ralgond commented 1 year ago

REQUIRED_PACKAGES = [ 'torch>=1.2.0', 'tqdm', 'scikit-learn', 'tensorflow' ]

我可以删掉setup.py中的tensorflow吗？

LiuHao-THU commented 1 year ago

我也想问...

xyz189411yt commented 1 year ago

basemodel.py里用到了tf的Callback相关组件。。以及callback.py里也用到了大量Callback类。感觉是在项目转torch的时候这部分忘记修改了？