Getting Started
Bring Your Own
Extending PyKEEN
Reference
pykeen.nn
Appendix
Generate random ranks from a given array of the number of candidates for each ranking task.
num_candidates (ndarray) – shape: s the number of candidates
ndarray
prefix_shape (Tuple[int, …]) – additional dimensions for broadcasted sampling
Tuple
int
seed (Union[None, int, Generator]) – the random seed
Union
None
Generator
dtype (Optional[Type[number]]) – the data type
Optional
Type
number
shape: dims + s an array of sampled rank values