The answer is to use `sagemaker.sklearn.processing.SKLearnProcessor` with a custom script that calls `sklearn.model_selection.StratifiedShuffleSplit`. This is correct because `StratifiedShuffleSplit` is specifically designed for stratified splitting for categorical distribution, ensuring that the relative frequencies of the 'region' feature are preserved across the training, validation, and test sets in the exact 70/20/10 ratio. On the AWS Certified Machine Learning Engineer Associate MLA-C01 exam, this question tests your understanding of how to combine SageMaker's managed processing infrastructure with standard scikit-learn methods for data preparation, rather than relying on built-in SageMaker split functions that do not support stratification. A common trap is choosing a generic `ScriptProcessor` or a built-in `SplitType` parameter, but neither can enforce class balance across multiple splits. Memory tip: think "Stratified = SKLearn" — whenever you need to preserve categorical distribution across splits in a SageMaker Processing job, reach for the `SKLearnProcessor` and `StratifiedShuffleSplit`.
MLA-C01 Data Preparation for Machine Learning Practice Question
This MLA-C01 practice question tests your understanding of data preparation for machine learning. Read the scenario carefully and evaluate each option against the stated constraints before committing to an answer. After answering, compare your reasoning against the explanation and wrong-answer breakdown below. Once you have made your selection, read the full explanation to reinforce the concept and understand why each distractor is designed to mislead on exam day.
A machine learning team is processing a large dataset in Amazon SageMaker using a processing job. The data is stored in S3 in CSV format. The team wants to split the data into training, validation, and test sets (70/20/10) while ensuring that the distribution of a categorical feature 'region' is preserved across splits. Which SageMaker SDK method should they use to write the output?
Answer the question above first, then reveal the full breakdown to understand why each option is right or wrong.
Correct answer & explanation
✓
Use sagemaker.sklearn.processing.SKLearnProcessor with a script that uses sklearn's StratifiedShuffleSplit
Option A is correct because `SKLearnProcessor` allows you to run a custom Python script that uses `sklearn.model_selection.StratifiedShuffleSplit`, which preserves the distribution of the categorical 'region' feature across the training, validation, and test splits. This is the only option that directly supports stratified splitting within a SageMaker processing job, ensuring the 70/20/10 ratio while maintaining class balance.
Key principle: Answer the scenario, not the keyword: identify the specific constraint before choosing the most familiar-sounding option.
Answer analysis
Option-by-option breakdown
For each option: why learners choose it and why it is or isn't the right answer here.
✓
Use sagemaker.sklearn.processing.SKLearnProcessor with a script that uses sklearn's StratifiedShuffleSplit
Why this is correct
StratifiedShuffleSplit ensures the 'region' distribution is maintained across splits.
Related concept
Read the scenario before looking for a memorised answer.
✗
Use sagemaker.xgboost.processing.XGBoostProcessor with a script that uses random split
Why it's wrong here
Random split does not preserve categorical distribution.
✗
Use sagemaker.processing.Processor.run() with a custom script that uses train_test_split
Why it's wrong here
train_test_split without stratification does not preserve distribution.
✗
Use sagemaker.processing.FrameworkProcessor with a script that uses pandas.sample
Why it's wrong here
pandas.sample does not support stratified splitting.
Common exam traps
Common exam trap: answer the scenario, not the keyword
The trap here is that candidates often confuse generic processing methods (like `Processor.run()` or `FrameworkProcessor`) with the specific processor that supports stratified splitting, or they assume `train_test_split` with a random state is sufficient for preserving categorical distributions, ignoring the need for stratification.
Detailed technical explanation
How to think about this question
Under the hood, `StratifiedShuffleSplit` works by first dividing the dataset into folds based on the categorical feature (e.g., 'region'), then sampling proportionally from each fold to create splits that mirror the original distribution. In a real-world scenario, if 'region' has imbalanced classes (e.g., 80% North America, 10% Europe, 10% Asia), a random split could accidentally place all Asia samples into the test set, leading to biased model evaluation; stratified splitting avoids this by ensuring each split contains the same class proportions.
KKey Concepts to Remember
Read the scenario before looking for a memorised answer.
Find the constraint that changes the correct option.
Eliminate answers that are true in general but not in this case.
TExam Day Tips
→Watch for words such as best, first, most likely and least administrative effort.
→Review why wrong options are wrong, not only why the correct option is correct.
Key takeaway
Answer the scenario, not the keyword: identify the specific constraint before choosing the most familiar-sounding option.
Real-world example
How this comes up in practice
A startup's cloud architect reviews their monthly bill and notices costs are higher than expected for a long-running batch job. Switching from on-demand instances to Reserved Instances — or using Spot/Preemptible VMs — can reduce compute costs by up to 72 %. Questions like this test whether you understand the tradeoffs between commitment, flexibility, and cost across cloud pricing models.
What to study next
Got this wrong? Here's your next step.
Identify which exam domain this question belongs to, review the core concept, then practise similar questions from the same domain.
Data Preparation for Machine Learning — This question tests Data Preparation for Machine Learning — Read the scenario before looking for a memorised answer..
What is the correct answer to this question?
The correct answer is: Use sagemaker.sklearn.processing.SKLearnProcessor with a script that uses sklearn's StratifiedShuffleSplit — Option A is correct because `SKLearnProcessor` allows you to run a custom Python script that uses `sklearn.model_selection.StratifiedShuffleSplit`, which preserves the distribution of the categorical 'region' feature across the training, validation, and test splits. This is the only option that directly supports stratified splitting within a SageMaker processing job, ensuring the 70/20/10 ratio while maintaining class balance.
What should I do if I get this MLA-C01 question wrong?
Identify which exam domain this question belongs to, review the core concept, then practise similar questions from the same domain.
What is the key concept behind this question?
Read the scenario before looking for a memorised answer.
About these practice questions
Courseiva creates original exam-style practice questions with explanations and wrong-answer analysis. It does not publish real exam questions, exam dumps, or protected exam content. Learn why practice questions differ from exam dumps →
Share a tip, memory trick, or ask about the reasoning behind this question. Do not post real exam questions, leaked content, braindumps, or copyrighted exam material. Comments are moderated and may be removed without notice.
This MLA-C01 practice question is part of Courseiva's free Amazon Web Services certification practice question bank. Courseiva provides original exam-style practice questions with explanations, topic-based practice, mock exams, readiness tracking, and study analytics to help learners prepare for the MLA-C01 exam.
Question Discussion
Share a tip, memory trick, or ask about the reasoning behind this question. Do not post real exam questions, leaked content, braindumps, or copyrighted exam material. Comments are moderated and may be removed without notice.
Sign in to join the discussion.