Source code for synapse.ml.causal.DoubleMLModel

# Copyright (C) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See LICENSE in project root for information.

import sys

if sys.version >= "3":
    basestring = str

from synapse.ml.causal._DoubleMLModel import _DoubleMLModel
from pyspark.ml.common import inherit_doc
import numpy as np


[docs]@inherit_doc
class DoubleMLModel(_DoubleMLModel):
[docs]    def getAvgTreatmentEffect(self):
        return self._java_obj.getAvgTreatmentEffect()

[docs]    def getConfidenceInterval(self):
        return list(self._java_obj.getConfidenceInterval())

[docs]    def getPValue(self):
        return self._java_obj.getPValue()