ShifuML / shifu

An end-to-end machine learning and data mining framework on Hadoop
https://github.com/ShifuML/shifu/wiki
Apache License 2.0
251 stars 109 forks source link

WDL: Customized Learning Rate for Wide and Deep #657

Open zhangpengshan opened 5 years ago

zhangpengshan commented 5 years ago

Current implementation only one learning rate for both wide and deep but the speed for both layer is different, we should support two learning rate.

For example, my testing in my data set wide layer is 5 learning rate but deep layer is 0.6.