Merge pull request #248 from ZJsheep/fix_bdeu

kunwuz · web-flow · commit 16d6db962c03 · 2025-12-22T02:35:02.000-05:00
Fix implementation of the structure prior term of the BDeu score
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`	`)`