매트릭스에 대한 numpy 합계

Question

데이터 포인트가 열 (시간 경과에 따른 농도) 인 행 (다른 pH에서 측정)이있는 매트릭스가 있습니다. 그래서 하나의 행은 하나의 pH에 ​​대해 다른 데이터 포인트로 구성됩니다.

데이터에 ODE를 맞추고 싶습니다. 그래서 비용 함수를 정의하고 모든 관측치의 제곱의 합을 계산하고 싶습니다. 이 행렬에 대한 sqares의 합계를 취하면 다음과 같이 작동합니다.

res = y - yhat      # calculate residuals 
ssq = np.diag(np.dot(res.T,res))  # sum over the diagonal 

이 맞습니까?

Answer 1

마지막 배열의 합계를 취하면 올바른 것입니다. 그러나 (오프 대각선 요소도 np.dot으로 계산되기 때문에)가 불필요하게 복잡 빠른입니다 : 당신은 각 실험에 대한 SSD를 원하는 경우

ssq = np.sum(res**2) 

, 당신이 할 수 있습니다

ssq = np.sum(res**2, axis=1)