def predict(self, X, **kwargs):
self.logger.info("Predicting ranks")
if isinstance(X, dict):
result = dict()
for n, scores in self.predict_scores(X, **kwargs).items():
predicted_rankings = scores_to_rankings(scores)
result[n] = predicted_rankings
return result
return ObjectRanker.predict(self, X, **kwargs)
def predict_for_scores(self, scores, **kwargs):
return ObjectRanker.predict_for_scores(self, scores, **kwargs)
def predict(self, X, **kwargs):
return ObjectRanker.predict(self, X, **kwargs)