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()