Fonction torch.corrcoef de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.corrcoefC'est une fonction dans PyTorch utilisée pour calculer la matrice des coefficients de corrélation de Pearson. Elle renvoie la matrice des coefficients de corrélation du tenseur d'entrée, qui est utilisée pour mesurer la corrélation linéaire entre les variables.

Définition de fonction

torch.corrcoef(input)

Exemple d'utilisation

Exemple

import torch

# Calculer la matrice des coefficients de corrélation
x = torch.tensor([[1, 2, 3, 4], [2, 3, 4, 5], [3, 4, 5, 6]])
print(Forme de la matrice d'entrée :, x.shape)
print(Entrée :)
print(x)

# Calculer la matrice des coefficients de corrélation
corr = torch.corrcoef(x)
print(Matrice des coefficients de corrélation :)
print(corr)

# Corrélation entre deux variables
a = torch.tensor([1, 2, 3, 4, 5])
b = torch.tensor([2, 4, 6, 8, 10])
# La forme doit être (2, n)
data = torch.stack([a, b])
corr2 = torch.corrcoef(data)
print(Coefficient de corrélation entre a et b :, corr2[0, 1])
# La sortie est proche de 1, indiquant une forte corrélation positive

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions