facebookresearch/StarSpace

loadBaseDocs() is called twice in query_predict utility

slvher opened this issue · 0 comments

If model is trained with trainMode=0 and fileFormat="labelDoc", and the query_predict utility is started with savedModel (instead of model.tsv), loadBaseDocs() will be called twice, at L33 and L41 respectively.
In such a situation, duplicated results will be returned by query_predict, due to labelDoc file is loaded twice.

$ ./bin/query_predict model/ss_model 6 data/basedoc.data

NOTE: "model/ss_model" is the savedModel name and "data/basedoc.data" is the basedoc file name.