Fix notebook issu

py-why · May 3, 2019 · 5fff32a · 5fff32a
1 parent de25eec
commit 5fff32a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)"
    ]
   },
   {