1import numpy as np
2
3def mse(y_true, y_pred):
4 y_true = np.array(y_true)
5 y_pred = np.array(y_pred)
6
7 return np.sum((y_true - y_pred) ** 2) / len(y_true)
8
9
10y_true = [3, -0.5, 2, 7]
11y_pred = [2.5, 0.0, 2, 8]
12
13print("MSE =", mse(y_true, y_pred))
STEP 00
準備開始
點「下一步」後,右邊會逐步畫出記憶體裡的變化。
BUILT-IN FUNCTION — print():內建函式物件;這裡只把結果輸出到終端機
terminal output
MSE = 0.375