predict_r_uncertain
- predict_r_uncertain(model, ht_batch, num_samples=5, slice_size=None)[source]
Forward pass using middle (relation) prediction for obtaining scores of all possible relations.
This method calculates the score for all possible relations for each (head, tail) pair, as well as an uncertainty quantification.
- Parameters
- Return type
- Returns
shape: (batch_size, num_relations) For each h-t pair, the scores for all possible relations.
This function delegates to
predict_uncertain_helper()
by usingpykeen.models.Model.score_r()
as thescore_method
.
Warning
This function sets the model to evaluation mode and all dropout layers to training mode.