PyTorch torch.corrcoef Funktion
PyTorch torch Referenzhandbuch
torch.corrcoefEs ist eine Funktion in PyTorch, die verwendet wird, um die Pearson-Korrelationskoeffizientenmatrix zu berechnen. Sie gibt die Korrelationskoeffizientenmatrix des Eingabetensors zurück, die verwendet wird, um die lineare Korrelation zwischen Variablen zu messen.
Funktionsdefinition
torch.corrcoef(input)
Anwendungsbeispiel
Beispiel
import torch
# Korrelationskoeffizientenmatrix berechnen
x = torch.tensor([[1, 2, 3, 4], [2, 3, 4, 5], [3, 4, 5, 6]])
print(Form der Eingabematrix:, x.shape)
print(Eingabe:)
print(x)
# Korrelationskoeffizientenmatrix berechnen
corr = torch.corrcoef(x)
print(Korrelationskoeffizientenmatrix:)
print(corr)
# Korrelation zwischen zwei Variablen
a = torch.tensor([1, 2, 3, 4, 5])
b = torch.tensor([2, 4, 6, 8, 10])
# Form muss (2, n) sein
data = torch.stack([a, b])
corr2 = torch.corrcoef(data)
print(Korrelationskoeffizient von a und b:, corr2[0, 1])
# Die Ausgabe liegt nahe bei 1, was auf eine starke positive Korrelation hindeutet
# Korrelationskoeffizientenmatrix berechnen
x = torch.tensor([[1, 2, 3, 4], [2, 3, 4, 5], [3, 4, 5, 6]])
print(Form der Eingabematrix:, x.shape)
print(Eingabe:)
print(x)
# Korrelationskoeffizientenmatrix berechnen
corr = torch.corrcoef(x)
print(Korrelationskoeffizientenmatrix:)
print(corr)
# Korrelation zwischen zwei Variablen
a = torch.tensor([1, 2, 3, 4, 5])
b = torch.tensor([2, 4, 6, 8, 10])
# Form muss (2, n) sein
data = torch.stack([a, b])
corr2 = torch.corrcoef(data)
print(Korrelationskoeffizient von a und b:, corr2[0, 1])
# Die Ausgabe liegt nahe bei 1, was auf eine starke positive Korrelation hindeutet
Andere Erweiterungen