birch icon indicating copy to clipboard operation
birch copied to clipboard

Published 20 hours ago • verified publisher icon castorini

Prediction score

Open fangguo1 opened this issue 4 years ago • 0 comments

Hi thanks for sharing birch!

I try to predict relevant sentences using the 'saved.msmarco_mb_1' model. One thing I am curious about is the prediction score I get from 'predictions = model(tokens_tensor, segments_tensor, mask_tensor)'. Each tuple in the 'predictions' does not sum to 1. Does it suppose to be a binary classification score?

fangguo1 avatar Dec 09 '20 17:12 fangguo1