yusanshi / news-recommendation

Implementations of some methods in news recommendation.
MIT License
241 stars 50 forks source link

有关于batch_size的问题 #31

Closed wjx0110 closed 2 years ago

wjx0110 commented 2 years ago

你好,我想问问你的代码里面batch_size的大小可以更改吗?还是说128已经是测试过以后更合适的值呢?

yusanshi commented 2 years ago

哈哈没怎么测试,你可以 16 32 64 128 256 ... 这样测一测

wjx0110 commented 2 years ago

你好,我想问问你train.py里面为什么每1000批以后要再计算一下??这里有点不太明白

yusanshi commented 2 years ago

可以说的更清楚一些吗?具体指哪行代码呢?

wjx0110 commented 2 years ago

mmexport1648962325059 就是这行代码?我不懂为什么1000batch之后要重新计算一下验证集

yusanshi commented 2 years ago

你好,主要是用于 Early stopping

https://github.com/yusanshi/news-recommendation/blob/master/src/train.py#L260-L277

wjx0110 commented 2 years ago

我想问问evaluate.py里面为什么还要有calculating vectors for news这些呢?在train.py里面不是已经计算过了吗?不是保存到ckpt文件里了嘛?直接计算auc不可以吗?

yusanshi commented 2 years ago

你好新的问题请开新 issue #34