search

motefly / DeepGBM

SIGKDD'2019: DeepGBM: A Deep Learning Framework Distilled by GBDT for Online Prediction Tasks
647 stars 135 forks source link

when _init_ DeepGBM,got a error #30

Open jaronn opened 3 years ago

jaronn commented 3 years ago

when init the class DeepFM in deepfm.py,got a error 微信图片_20201112104749

jaronn commented 3 years ago

got it！I change the 1st para "sum(feature_size)" in nn.Embedding() to "int(sum(feature_size))", it works. this para must be int.