Want to get the top 10 translation results

[PPPNut]

I want to get the best 10 translations through the search algorithm, but I don't know how to do it.

[hzhwcmhf]

The current implementation does not support this function. You may need to modify dag_search.

First,

https://github.com/thu-coai/DAG-Search/blob/d5056b8da20495be6794967a718f2a0a4f5df227/python/dag_search.pyx#L159-L167

L160 specifies the number of beams after sorting and filtering, it will be 1 when step is the last position. You may change it to 10 if you want 10 translations.

Second,

https://github.com/thu-coai/DAG-Search/blob/d5056b8da20495be6794967a718f2a0a4f5df227/python/dag_search.pyx#L181-L187

This function obtains the paths of the beams. It currently only obtains the path of the best beam (i.e., beams[i * SearchBeam.max_pos]), and save it to result. You may change it to beams[i * SearchBeam.max_pos + j] (j from 0 to 9) to get the other path.

You will need to compile the codes and reinstall dag_search by

cd dag_search
python3 setup.py build_ext --inplace
pip install -e .

I am sorry that I am busy in my work and does not have time to add this function recently. If you have any problem, please reply in this issue.

[PPPNut]

I have spent a long time trying to modify it, but without success. May I ask if the author has time to add this feature? I would be extremely grateful. 🙏

[hzhwcmhf]

I will try to add this feature in two weeks.

[PPPNut]

Thank you sincerely.

[hzhwcmhf]

I push an update to the nbest branch.

git fetch
git checkout nbest
git submodule update
cd dag_search
python3 setup.py build_ext --inplace
pip install -e .

You can then add --decode-final-beamsize 10 --nbest 10 in beamsearch to obtain 10 best candidates.

Please let me know whether it works well. I will merge it into main if no bugs exist. @PPPNut

[PPPNut]

Thank you very much for your guidance and advice. I will promptly provide you with feedback if there are any questions.