A/B testing models in ML Python - Cheat Sheet & Quick Revision

Recall & Review

beginner

What is A/B testing in the context of machine learning models?

A/B testing is a method to compare two versions of a machine learning model by running them simultaneously on different groups of users to see which performs better.

beginner

Why do we split users into groups during A/B testing?

We split users into groups to ensure each model is tested fairly and independently, so the results show which model works better without bias.

intermediate

What metric would you use to decide the winner in an A/B test for a recommendation model?

You might use metrics like click-through rate, conversion rate, or accuracy depending on the goal of the recommendation model.

beginner

How does A/B testing help in improving machine learning models?

A/B testing helps by providing real user feedback on different models, allowing data-driven decisions to pick the best performing model.

intermediate

What is a common risk when running A/B tests on machine learning models?

A common risk is that the test groups are not representative or large enough, which can lead to wrong conclusions about model performance.

What does A/B testing compare in machine learning?

A Two datasets for training

B Two features of the same model

C Two different models on separate user groups

D Two hyperparameters in one model

A/B testing compares two different models by running them on separate user groups to see which performs better.

Why is random assignment important in A/B testing?

A To speed up training

B To ensure fairness and reduce bias

C To increase model complexity

D To reduce data size

Random assignment ensures that each group is similar, making the test fair and unbiased.

Which metric is NOT typically used to evaluate A/B tests for models?

A Click-through rate

B Conversion rate

C User engagement

D Training loss

Training loss is used during model training, not for evaluating live A/B test results.

What should you do if A/B test results are inconclusive?

A Increase sample size and test duration

B Stop testing immediately

C Pick the model randomly

D Ignore the test and deploy old model

Increasing sample size and test duration helps get clearer results.

What is a key benefit of A/B testing models in production?

A Real user feedback on model performance

B Faster model training

C Reduced data collection

D Simpler model design

A/B testing provides real user feedback, which is valuable for understanding model impact.

Explain how A/B testing works for comparing two machine learning models.

Think about how you would test two recipes by giving each to different friends and seeing which they like more.

Describe common challenges and risks when performing A/B testing on models.

Consider what could go wrong if you test a new product with only a few people or the wrong questions.

Practice

1. What is the main purpose of A/B testing models in machine learning?

easy

A. To compare two models by testing them on separate groups

B. To train a single model on all data

C. To reduce the size of the dataset

D. To increase the number of features in the model

Solution

1. Step 1: Understand the goal of A/B testing

   A/B testing is used to compare two different models or versions by testing them separately.

2. Step 2: Identify the correct purpose

   The purpose is to find which model performs better by testing on separate groups, not to train on all data or change dataset size.

3. Final Answer:

   To compare two models by testing them on separate groups -> Option A

4. Quick Check:

   A/B testing = compare models [OK]

Hint: A/B testing means comparing two models on different groups [OK]

Common Mistakes:

• Thinking A/B testing trains one model on all data
• Confusing A/B testing with feature engineering
• Believing A/B testing reduces dataset size

2. Which of the following is the correct way to split data for A/B testing?

easy

A. Train both models on the entire dataset simultaneously

B. Use the same data for both models without splitting

C. Split data randomly into two groups and test each model separately

D. Split data by sorting values and assigning first half to model A

Solution

1. Step 1: Recall proper data splitting for A/B testing

   Data should be split randomly to avoid bias and ensure fair comparison.

2. Step 2: Evaluate options

   Only random splitting into two groups is correct; using same data or sorted split causes bias.

3. Final Answer:

   Split data randomly into two groups and test each model separately -> Option C

4. Quick Check:

   Random split = fair test [OK]

Hint: Always split data randomly for fair A/B testing [OK]

Common Mistakes:

• Using same data for both models
• Splitting data by sorting values
• Training both models on full data

3. Consider this Python code for A/B testing two models on separate data splits:

from sklearn.model_selection import train_test_split
from sklearn.linear_model import LogisticRegression
from sklearn.metrics import accuracy_score

X = [[1], [2], [3], [4], [5], [6]]
y = [0, 0, 0, 1, 1, 1]

X_train_A, X_test_A, y_train_A, y_test_A = train_test_split(X[:3], y[:3], test_size=1, random_state=42)
X_train_B, X_test_B, y_train_B, y_test_B = train_test_split(X[3:], y[3:], test_size=1, random_state=42)

model_A = LogisticRegression().fit(X_train_A, y_train_A)
model_B = LogisticRegression().fit(X_train_B, y_train_B)

pred_A = model_A.predict(X_test_A)
pred_B = model_B.predict(X_test_B)

acc_A = accuracy_score(y_test_A, pred_A)
acc_B = accuracy_score(y_test_B, pred_B)

print(acc_A, acc_B)

What will be the printed output?

medium

A. 0.0 0.0

B. 1.0 1.0

C. 1.0 0.0

D. 0.0 1.0

Solution

1. Step 1: Analyze data splits and training

   Each model trains on 2 samples and tests on 1 sample. The test samples are [3] for A and [6] for B with labels 0 and 1 respectively.

2. Step 2: Predict and calculate accuracy

   LogisticRegression will perfectly separate these simple points, so predictions match true labels, giving accuracy 1.0 for both.

3. Final Answer:

   1.0 1.0 -> Option B

4. Quick Check:

   Perfect split and simple data = accuracy 1.0 [OK]

Hint: Small simple splits often give perfect accuracy [OK]

Common Mistakes:

• Assuming accuracy is zero due to small data
• Confusing train and test splits
• Ignoring random_state effect

4. You run an A/B test comparing two models but notice both groups have very different user ages. What is the likely problem?

medium

A. Data was not split randomly, causing bias

B. Models are overfitting the training data

C. Test size was too large

D. Models have too many features

Solution

1. Step 1: Identify issue from uneven user ages

   Unequal age distribution means groups are not similar, indicating non-random splitting.

2. Step 2: Understand impact on A/B testing

   Non-random splits cause bias, making test results unreliable.

3. Final Answer:

   Data was not split randomly, causing bias -> Option A

4. Quick Check:

   Unequal groups = non-random split [OK]

Hint: Unequal groups mean split was not random [OK]

Common Mistakes:

• Blaming overfitting for group differences
• Thinking test size causes group bias
• Confusing feature count with data split

5. You want to run an A/B test comparing two recommendation models on a website. Which approach best ensures a fair test?

hard

A. Assign users based on their signup date to different models

B. Show both models to all users and average their recommendations

C. Train both models on the same user data and pick the one with higher training accuracy

D. Randomly assign users to Model A or Model B groups and collect click data separately

Solution

1. Step 1: Understand fair A/B testing for users

   Random assignment ensures groups are similar and results are unbiased.

2. Step 2: Evaluate other options

   Showing both models to all users or assigning by signup date causes bias; training accuracy alone is not enough.

3. Final Answer:

   Randomly assign users to Model A or Model B groups and collect click data separately -> Option D

4. Quick Check:

   Random user assignment = fair A/B test [OK]

Hint: Random user assignment ensures unbiased A/B test [OK]

Common Mistakes:

• Showing both models to same users
• Using signup date for group assignment
• Choosing model by training accuracy only