Update README.md

README.md

@@ -132,6 +132,18 @@ custom_rankings = model.rank(query, documents, prompt=instruction)
 
 # Use Identity to return yes_logit - no_logit instead of P(yes).
 margins = model.predict(pairs, activation_fn=torch.nn.Identity())
+
+print(f"scores: {scores}")
+print(f"rankings: {rankings}")
+print(f"custom_scores: {custom_scores}")
+print(f"custom_rankings: {custom_rankings}")
+print(f"margins: {margins}")
+
+# scores: [9.9982053e-01 4.7850945e-06]
+# rankings: [{'corpus_id': 0, 'score': 0.99982053, 'text': 'The capital of China is Beijing.'}, {'corpus_id': 1, 'score': 4.7850945e-06, 'text': 'Gravity attracts bodies toward one another.'}]
+# custom_scores: [9.9956948e-01 1.4883849e-04]
+# custom_rankings: [{'corpus_id': 0, 'score': 0.9995695}, {'corpus_id': 1, 'score': 0.00014883849}]
+# margins: [  8.625 -12.25 ]
 ```
 
 Inputs must be ordered as `(query, document)`. By default, queries are