5

2 次元の行列が与えられた場合、対応する共分散行列を計算したいと思います。

この操作を容易にする Nd4j に含まれるメソッドはありますか?

たとえば、次の行列から計算された共分散行列

1  2
8 12

ここでNd4jを使用して構築されました:

INDArray array1 = Nd4j.zeros(2, 2);  
array1.putScalar(0, 0, 1);
array1.putScalar(0, 1, 2);
array1.putScalar(1, 0, 8);
array1.putScalar(1, 1, 12);

する必要があります

24.5  35.0
35.0  50.0

これは、次のように pandas の DataFrame のcovメソッドを使用して簡単に実行できます。

>>> pandas.DataFrame([[1, 2],[8, 12]]).cov()
      0     1
0  24.5  35.0
1  35.0  50.0

Nd4j を使用してこれを行う方法はありますか?

4

1 に答える 1