Data Science, Causal Inference, Machine Learning, Optimization, Notes From Industry.
return untreated Again, we do this for each product: untreated_1 = extract_untreated_effect(n, X_train_1, model_1) untreated_2 = extract_untreated_effect(n, X_train_2, model_2) untreated_3 = extract_untreated_effect(n, X_train_3, model_3) We can now calculate the treatment effect for the full range