6967cc5df1801b8f5a01110417b38495896d75e0,src/spn/structure/leaves/cltree/Inference.py,,cltree_likelihood,#Any#Any#Any#,9

Before Change

from spn.algorithms.Inference import add_node_likelihood

def cltree_likelihood(node, data=None, dtype=np.float64):
    raise ValueError("Not Implemented")

def add_cltree_inference_support():
    add_node_likelihood(CLTree, cltree_likelihood)

After Change

import numpy as np

def cltree_likelihood(node, data=None, dtype=np.float64):
    probs = np.zeros(data.shape[0], dtype=dtype)

    for feature in range(0, node.n_features):
        parent = node.tree[feature]
        if parent == -1:
            probs = probs + node.log_factors[feature, data[:,node.scope[feature]],0]
        else:
            probs = probs + node.log_factors[feature, data[:,node.scope[feature]], data[:,node.scope[parent]]]

    return np.exp(probs.reshape(data.shape[0],1))
    

Instances