Fix implementation of BDeu score

ZJsheep · web-flow · commit d486643e95ed · 2025-12-16T17:31:13.000+08:00
Fix calculation of vm to use the number of columns (number of variables) instead of rows (number of samples).

Signed-off-by: Zhou Jiayang &lt;73184640+ZJsheep@users.noreply.github.com&gt;
diff --git a/causallearn/score/LocalScoreFunction.py b/causallearn/score/LocalScoreFunction.py
@@ -148,7 +148,7 @@ def local_score_BDeu(Data: ndarray, i: int, PAi: List[int], parameters=None) ->
 
     BDeu_score = 0
     # first term
-    vm = Data.shape[0] - 1
+    vm = Data.shape[1] - 1
     BDeu_score += len(PAi) * np.log(structure_prior / vm) + (vm - len(PAi)) * np.log(
         1 - (structure_prior / vm)
     )

Original file line number	Diff line number	Diff line change
`@@ -148,7 +148,7 @@ def local_score_BDeu(Data: ndarray, i: int, PAi: List[int], parameters=None) ->`
`148`	`148`
`149`	`149`	`BDeu_score = 0`
`150`	`150`	`# first term`
`151`		`- vm = Data.shape[0] - 1`
	`151`	`+ vm = Data.shape[1] - 1`
`152`	`152`	`BDeu_score += len(PAi) * np.log(structure_prior / vm) + (vm - len(PAi)) * np.log(`
`153`	`153`	`1 - (structure_prior / vm)`
`154`	`154`	`)`