diff --git a/causallearn/score/LocalScoreFunction.py b/causallearn/score/LocalScoreFunction.py index b87ffe4..2f2d7a1 100644 --- a/causallearn/score/LocalScoreFunction.py +++ b/causallearn/score/LocalScoreFunction.py @@ -682,7 +682,7 @@ def local_score_marginal_general( Data: ndarray, Xi: int, PAi: List[int], parameters=None ) -> float: """ - Calculate the local score by negative marginal likelihood + Calculate the local score by negative marginal log-likelihood based on a regression model in RKHS Parameters @@ -784,7 +784,7 @@ def local_score_marginal_multi( Data: ndarray, Xi: int, PAi: List[int], parameters: Dict[str, Any] ) -> float: """ - Calculate the local score by negative marginal likelihood + Calculate the local score by negative marginal log-likelihood based on a regression model in RKHS for variables with multi-variate dimensions @@ -887,4 +887,4 @@ def local_score_marginal_multi( nargout=2, ) score = nlml # negative log-likelihood - return score \ No newline at end of file + return score