调参日志:记录参数、训练集得分、测试集(提交答案)得分 - ShenDezhou/rct-intel-hw-recommend GitHub Wiki

  • #1#3, #2#4 shows by 有l2_norm(即不设--embed_l2=1e-6参数)更好
  • #1#2 shows by 减少epoch+增大lr更好
  • #1#5 shows by 减少batch size几乎没影响
  • #1#6 shows by 减小embed dim更好
  • #5#7 shows by 减少neg来平衡数据更糟
  • #1#8 shows by 增加来平衡数据更糟,平衡数据没有收益

1.--epochs=10 --batch_size=1024 --embed_dim=10 --learning_rate=1e-2

线上训练集: {'read_comment': 0.7933732360130826, 'like': 0.7580773920201094, 'click_avatar': 0.8567413676355881, 'forward': 0.864511659963672} Weighted uAUC: 0.802572

线上测试集: 0.614079

2.--epochs=100 --batch_size=1024 --embed_dim=10 --learning_rate=1e-3

线上Train: {'read_comment': 0.9296389102634225, 'like': 0.9276057828706135, 'click_avatar': 0.9763918319179635, 'forward': 0.9921951744441916} Weighted uAUC: *0.944635*

线上Test: 0.581364

3.--epochs=10 --batch_size=1024 --embed_dim=10 --learning_rate=1e-2 --embed_l2=1e-6

线上Train: {'read_comment': 0.5728769691240823, 'like': 0.5701266322666655, 'click_avatar': 0.6763979779112091, 'forward': 0.6589805821336259} Weighted uAUC: *0.601366*

线上Test: 0.572742

4.--epochs=100 --batch_size=1024 --embed_dim=10 --learning_rate=1e-3 --embed_l2=1e-6

线上Train: {'read_comment': 0.589317704686516, 'like': 0.5791047343396772, 'click_avatar': 0.6916181607551332, 'forward': 0.6714297988740285} Weighted uAUC: *0.614925*

线上Test: 0.550584

5.--epochs=10 --batch_size=8 --embed_dim=10 --learning_rate=1e-2

线上Train: {'read_comment': 0.7170037704597131, 'like': 0.7179916952979987, 'click_avatar': 0.830932173337868, 'forward': 0.8569557624429204} Weighted uAUC: 0.754081

线上Test: 0.614417

6.--epochs=10 --batch_size=1024 --embed_dim=32 --learning_rate=1e-2

线上Train: {'read_comment': 0.929857967278868, 'like': 0.9109560878170173, 'click_avatar': 0.9712593062700721, 'forward': 0.9767509915612788} Weighted uAUC: 0.937157

线上Test: 0.602205

7. 平衡数据(减少neg到pos),--epochs=10 --batch_size=8 --embed_dim=10 --learning_rate=1e-2

线上Train: {'read_comment': 0.7333809135325238, 'like': 0.7306940034156933, 'click_avatar': 0.8442448791976356, 'forward': 0.8246810086586831} Weighted uAUC: 0.763878

线上Test: 0.592053

8. 平衡数据(复制pos到neg),--epochs=10 --batch_size=1024 --embed_dim=10 --learning_rate=1e-2

线上Train: {'read_comment': 0.9661392482912947, 'like': 0.9877125054445267, 'click_avatar': 0.9991942156567207, 'forward': 0.9996995867431433} Weighted uAUC: 0.982578

线上Test: 0.557775