GanjinZero / RRHF

[NIPS2023] RRHF & Wombat
780 stars 49 forks source link

有关IMDB数据集的问题 #46

Open stevie1023 opened 9 months ago

stevie1023 commented 9 months ago

您好~想请问一下您在附录中有关IMDB情感分类数据集的实验细节。因为数据集中对每个query只包含单条样本,我们尝试使用imdb-gpt2进行采样扩充出另一条样本,可是效果不是很理想,所以想请问一下您是如何构建RRHF的训练样本对的呢

祝好~:)

Yuanhy1997 commented 9 months ago

我们就是仿照之前的工作,给定imbd测试样例的prefix,做去decoding产生不同的样本。请问你是如何进行采样扩充的呢,还有效果不理想具体是指什么呢?

stevie1023 commented 9 months ago

我们也是给定imbd测试样例的prefix,然后使用imdb-gpt2做去decoding产生样本,但是生成的样本质量(打分)不高,且有很多生成为空的情况。请问您是用什么模型采样样本的呢?还是说是因为我们的decoding参数设置不太合理:(