Skip to content

Commit 5399515

Browse files
committed
final updates
1 parent 9ea1fed commit 5399515

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

causallearn/score/LocalScoreFunction.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -34,9 +34,9 @@ def local_score_BIC(Data: ndarray, i: int, PAi: List[int], parameters=None) -> f
3434
if len(PAi) == 0:
3535
return n * np.log(cov[i, i])
3636

37-
yX = np.mat(cov[np.ix_([i], PAi)])
38-
XX = np.mat(cov[np.ix_(PAi, PAi)])
39-
H = np.log(cov[i, i] - yX * np.linalg.inv(XX) * yX.T)
37+
yX = cov[np.ix_([i], PAi)]
38+
XX = cov[np.ix_(PAi, PAi)]
39+
H = np.log(cov[i, i] - yX @ np.linalg.inv(XX) @ yX.T)
4040

4141
return n * H + np.log(n) * len(PAi) * lambda_value
4242

@@ -68,9 +68,9 @@ def local_score_BIC_from_cov(
6868
if len(PAi) == 0:
6969
return n * np.log(cov[i, i])
7070

71-
yX = np.mat(cov[np.ix_([i], PAi)])
72-
XX = np.mat(cov[np.ix_(PAi, PAi)])
73-
H = np.log(cov[i, i] - yX * np.linalg.inv(XX) * yX.T)
71+
yX = cov[np.ix_([i], PAi)]
72+
XX = cov[np.ix_(PAi, PAi)]
73+
H = np.log(cov[i, i] - yX @ np.linalg.inv(XX) @ yX.T)
7474

7575
return n * H + np.log(n) * len(PAi) * lambda_value
7676

0 commit comments

Comments
 (0)