Computes the most likely components for each of the provided datapoints.
- Return type
A tensor of shape
[num_datapoints]with the indices of the most likely components.
predict_proba()to obtain probabilities for each component instead of the most likely component only.
When calling this function in a multi-process environment, each process receives only a subset of the predictions. If you want to aggregate predictions, make sure to gather the values returned from this method.