Fixes calculuation of hat values

cvxgrp · Dec 12, 2024 · 918a951 · 918a951
1 parent 0abd6cf
commit 918a951
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/utils/sherlock_script.py b/utils/sherlock_script.py
@@ -61,8 +61,8 @@
 L = state.betas.shape[0]
 oos = np.empty(L)
 ins = np.empty(L)
-y_hat_test = X_test @ state.betas.T
-y_hat_train = X_train @ state.betas.T
+y_hat_test = ad.diagnostic.predict(X_test, state.betas, state.intercepts)
+y_hat_train = ad.diagnostic.predict(X_train, state.betas, state.intercepts)
 for i in range(L):
     oos[i] = loss(torch.from_numpy(y_hat_test[:, i]), torch.from_numpy(y_test))
     ins[i] = loss(torch.from_numpy(y_hat_train[:, i]), torch.from_numpy(y_train))