search

huawei-noah / vega

AutoML tools chain
http://www.noahlab.com.hk/opensource/vega/
Other
842 stars 175 forks source link

Issue with processing multi-value features #113

Open christine415 opened 3 years ago

christine415 commented 3 years ago

I am doing a CTR prediction project, whose dataset has multi-value features as input. How can I write a Dataset class with using Pytorch as backend. For example, uid history_click history_category nid label 00 aa, bb news, sports cc 0

christine415 commented 3 years ago

Can anyone help with this question?

I am doing a CTR prediction project, whose dataset has multi-value features as input. How can I write a Dataset class with using Pytorch as backend. For example, uid history_click history_category nid label 00 aa, bb news, sports cc 0

zhangjiajin commented 3 years ago

@christine415

There is an issue #106 related to your question. Vega has two algorithms related to CTR (Avazu) data. For details, see the data organization method of Avazu.