wellcometrust / grants_tagger

Tag grants with MeSH and other tags
MIT License
14 stars 4 forks source link

Onlytop_k not working #193

Closed aCampello closed 2 years ago

aCampello commented 2 years ago

When predicting mesh-xlinear with only_topk the value of 20 is always used, regardless of which value is set up. Therefore we can only output 20 predictions for each datapoint, which is problematic in high recall set-ups.

The reason for that is because this parameter should be passed to prediction.