in causalml/inference/tree/uplift.pyx [0:0]
def __init__(self, classes_, col=-1, value=None, trueBranch=None, falseBranch=None, results=None, summary=None,
maxDiffTreatment=None, maxDiffSign=1., nodeSummary=None, backupResults=None, bestTreatment=None,
upliftScore=None, matchScore=None):
self.classes_ = classes_
self.col = col
self.value = value
self.trueBranch = trueBranch
self.falseBranch = falseBranch
self.results = results # None for nodes, not None for leaves
self.summary = summary
# the treatment with max( |p(y|treatment) - p(y|control)| )
self.maxDiffTreatment = maxDiffTreatment
# the sign for p(y|maxDiffTreatment) - p(y|control)
self.maxDiffSign = maxDiffSign
self.nodeSummary = nodeSummary
self.backupResults = backupResults
self.bestTreatment = bestTreatment
self.upliftScore = upliftScore
# match actual treatment for validation and testing
self.matchScore = matchScore