Data Science, Product Analytics, Ab Testing, Simulation, Deep Dives.
retention_model.predict( experiment_df[['retention', 'treatment_eq_1', 'num_family_members']]\ .rename(columns = {'treatment_eq_1': 'treatment'})) experiment_df['retention_proba_control'] = retention_model.predict( experiment_df[['retention', 'treatment_eq_0', 'num_family_members']]\