Created by: elenaermakova
- Added files for mltrack ranking:
-
reader
mltrack_reader.py
to read and to preprocess raw data -
config file
mltrack_ranker.json
for model training -
ndcg.py
metric for scoring observed predictions -
notebook
mltrack_submit.ipynb
to get submission file
- Bug in return_probas condition:
If
self.return_probas
is True, then error appears in the following line becausepred
dimensionality in this case is equal to 1. Moreover,predictions
is a vector containing probabilities of the last class which doesn't make sense in case of non-binary classification. https://github.com/deepmipt/DeepPavlov/blob/0214bd4ca2a676057d287a5011a20fb7ee2ddaeb/deeppavlov/models/bert/bert_ranker.py#L101