Created by: elenaermakova
- Added files for mltrack ranking:
-
reader
mltrack_reader.pyto read and to preprocess raw data -
config file
mltrack_ranker.jsonfor model training -
ndcg.pymetric for scoring observed predictions -
notebook
mltrack_submit.ipynbto get submission file
- Bug in return_probas condition:
If
self.return_probasis True, then error appears in the following line becausepreddimensionality in this case is equal to 1. Moreover,predictionsis 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