How to add non-categorical feature to deepfm?

rixwew / pytorch-fm

Factorization Machine models in PyTorch

MIT License

1.04k stars 225 forks source link

How to add non-categorical feature to deepfm? #16

Closed OptimusPrimeCao closed 4 years ago

OptimusPrimeCao commented 4 years ago

How to add non-categorical features such as human age to deepfm model? It seems the deepfm implementation only supports categorical feature?

rixwew commented 4 years ago

Currently DeepFM implementation only supports categorical feature, so you need to do data binning for numeric features. I hope to support non-categorical features in the future.