geek-ai/irgan

How to generate dns_nn.model?

kingofoz opened this issue · 1 comments

Hi @wabyking @wabyking
I found there is an initial model in your github: dns_nn.model.
Can you tell me how to generate(train) it?
Another question is, can you give me some instructions about how to train IRGAN? Seems you just show the method to evaluate the stored models.

Thanks,
Yingjun

Hi @kingofoz

For DNS model you can run ltr-gan-pointwise/ltr_dns_nn.py;
for LTR-pointwise model you can run ltr-gan-pointwise/ltr_gan_d_nn_g_nn.py;
for LTR-pairwise model you can run ltr-gan-pairwsie/ltr_gan_d_nn_g_nn.py.

Thanks,
Yu