search

jrzaurin / pytorch-widedeep

A flexible package for multimodal-deep-learning to combine tabular data with text and images using Wide and Deep models in Pytorch
Apache License 2.0
1.27k stars 188 forks source link

about dataset #219

Closed xylovezxy closed 1 month ago

xylovezxy commented 1 month ago

Hi！Do you have any example code for using the Movielens dataset for CTR prediction？

jrzaurin commented 1 month ago

is not CTR per se but you have examples with the Movilens dataset here: https://github.com/jrzaurin/pytorch-widedeep/blob/master/examples%2Fnotebooks%2F19_wide_and_deep_for_recsys_pt1.ipynb

(And part 2 of that notebook)