Función torch.corrcoef de PyTorch

Manual de referencia de Pytorch torch

torch.corrcoefEs una función en PyTorch para calcular la matriz de coeficientes de correlación de Pearson. Devuelve la matriz de coeficientes de correlación del tensor de entrada, utilizada para medir la correlación lineal entre variables.

Definición de funciones

torch.corrcoef(input)

Ejemplo de uso

Ejemplos

import torch

# Calcular la matriz de coeficientes de correlación
x = torch.tensor([[1, 2, 3, 4], [2, 3, 4, 5], [3, 4, 5, 6]])
print(Forma de la matriz de entrada:, x.shape)
print("Entrada:")
print(x)

# Calcular la matriz de coeficientes de correlación
corr = torch.corrcoef(x)
print(Matriz de coeficientes de correlación:)
print(corr)

# Correlación entre dos variables
a = torch.tensor([1, 2, 3, 4, 5])
b = torch.tensor([2, 4, 6, 8, 10])
# La forma debe ser (2, n)
data = torch.stack([a, b])
corr2 = torch.corrcoef(data)
print(Coeficiente de correlación entre a y b:, corr2[0, 1])
# La salida cercana a 1 indica una fuerte correlación positiva

Manual de referencia de Pytorch torch

Otras extensiones