Data Science, Machine Learning, Causal Inference, AI, Linear Programming.
DML model using flexible stage 1 models dml2 = LinearDML(model_y=LGBMRegressor(), model_t=LGBMClassifier(), discrete_treatment=True) dml2.fit(y2, T=T2, X=X2, W=None) # Train DML model using flexible stage 1 models dml3 = LinearDML(model_y=LGBMRegressor(), model_t=LGBMClassifier(),