From 5fff32a616e54293a5ab22726f32b69bdebd5cd7 Mon Sep 17 00:00:00 2001 From: Keith Battocchi Date: Thu, 2 May 2019 18:40:11 -0400 Subject: [PATCH] Fix notebook issu --- notebooks/Double Machine Learning Examples.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/notebooks/Double Machine Learning Examples.ipynb b/notebooks/Double Machine Learning Examples.ipynb index 41b07f81c..237a4afa0 100644 --- a/notebooks/Double Machine Learning Examples.ipynb +++ b/notebooks/Double Machine Learning Examples.ipynb @@ -956,7 +956,7 @@ "source": [ "est = DMLCateEstimator(model_y=MultiTaskElasticNetCV(cv=3),model_t=MultiTaskElasticNetCV(cv=3))\n", "est.fit(Y, T, X, W)\n", - "te_pred=est.effect(X_test)" + "te_pred=est.const_marginal_effect(X_test)" ] }, {