Data Science, Machine Learning, Causal Inference, AI, Linear Programming.
discrete_treatment=True) dml3.fit(y3, T=T3, X=X3, W=None) When we plot the actual vs estimated CATE, we see that the model does a reasonable job. # Create a figure and subplots fig, axes = plt.subplots(1, 3, figsize=(15, 5)) # Plot scatter plots on each subplot sns.scatterplot(x=dml1.effect(X1),