Databricks Machine Learning Associate study materials
Databricks Machine Learning Associate study resources

Databricks Machine Learning Associate study plan

Databricks Machine Learning Associate · Give each week a clear focus.

8 weeks · About 7 hours per week

Week 1

Databricks Machine Learning

  • Describe how the Databricks Machine Learning workspace organizes notebooks, clusters, repos, and experiments, and diagram that relationship from memory

  • Configure a single-node and a multi-node cluster, then explain which settings control runtime version, autoscaling, and library installation

  • Install a Python ML library at both the cluster and notebook scope, then identify which scope wins when versions conflict

  • Run a minimal training script on a Databricks cluster and read the driver logs to diagnose one intentionally introduced failure

  • Compare the Databricks Runtime ML variants and select the correct one for a GPU-based deep learning task

Your checkpoint

A one-page architecture cheat sheet covering workspace objects, cluster configuration knobs, and library scoping precedence, plus a working notebook that trains and logs a scikit-learn model end to end

Week 2

Databricks Machine Learning

  • Register a trained model in the model registry and promote it through Staging to Production using registry stages and aliases

  • Create a feature table, then look up training features by primary key and timestamp to avoid label leakage

  • Explain how Unity Catalog governs models, features, and experiments across workspaces, and apply a grant to restrict access to one registered model

  • Trace a model from notebook run to registry version to serving endpoint by following its identifiers

  • Identify which Databricks ML governance controls are workspace-scoped versus account-scoped

Your checkpoint

A registered model with at least two versions, an attached description and tag set, and a written access-control matrix showing which principal can read, write, and manage it

Week 3

Model Development

  • Engineer numeric, categorical, and text features inside a Pandas or Spark DataFrame and handle nulls, skew, and high-cardinality columns deliberately

  • Split data into train, validation, and test sets without temporal or group leakage, and justify the splitting strategy in writing

  • Train baseline and tuned models for both classification and regression, and select evaluation metrics appropriate to class imbalance

  • Run a hyperparameter sweep with MLflow tracking and parallelize trials across a multi-node cluster

  • Detect and mitigate overfitting by comparing train versus validation learning curves

Your checkpoint

A notebook that trains at least three model families on one dataset, logs all runs to MLflow, and includes a metrics comparison table naming the selected model and the reason it won

Week 4

Model Development

  • Implement a reproducible preprocessing pipeline that is fit only on training data and reused unchanged at inference time

  • Tune hyperparameters with an automated search and record the search space, sampler, and best trial in MLflow

  • Evaluate a model on a held-out test set and produce a confusion matrix plus per-class precision and recall

  • Diagnose a model with feature importance or permutation importance and remove at least one feature that adds no signal

  • Package the final model with its preprocessing steps into a single artifact that accepts raw input

Your checkpoint

A packaged end-to-end pipeline artifact plus a short written model card documenting the dataset, metrics, known limitations, and the exact command to retrain it

Week 5

ML Workflows

  • Structure an MLflow experiment hierarchy with nested runs and consistent tag naming so results are comparable across team members

  • Log parameters, metrics, artifacts, and a model signature, then reload the logged model in a fresh notebook and confirm predictions match

  • Reproduce a prior run from its logged metadata alone, without reusing the original notebook state

  • Compare runs programmatically in a DataFrame and select the best configuration by a stated rule

  • Describe how Delta Lake tables support reproducible training data through versioning and time travel

Your checkpoint

A reproducible training notebook that a second person could run unchanged to obtain the same logged metrics, plus a run-comparison table of at least ten experiments

Week 6

ML Workflows

  • Schedule a multi-task Databricks job that chains data preparation, training, and evaluation with explicit task dependencies

  • Parameterize a notebook so the same code runs for dev and prod by passing widget values from a job

  • Configure retries, timeouts, and failure notifications on a job and explain when each matters

  • Query the job run history to confirm a scheduled run succeeded and to retrieve its output metrics

  • Assemble a CI-style workflow that validates data quality before training begins

Your checkpoint

A scheduled multi-task job that runs unattended on a trigger, produces a logged model version, and whose run history you can walk through task by task

Week 7

Model Deployment

  • Deploy a registered model to a real-time serving endpoint and send a scored request to it

  • Perform batch inference with a Spark job and write predictions to a Delta table partitioned for downstream consumption

  • Choose between real-time serving, batch scoring, and streaming inference for three described scenarios and justify each choice

  • Explain how endpoint scaling, version promotion, and rollback work when a bad model version reaches production

  • Implement a basic drift or data-quality check on incoming requests and describe what should trigger a retrain

Your checkpoint

One live serving endpoint returning predictions plus one batch inference job writing to a Delta table, documented with request examples, expected latency, and a rollback procedure

Week 8

Databricks Machine Learning & ML Workflows & Model Development & Model Deployment

  • Re-derive from memory the full lifecycle from raw data to a governed, deployed model and identify weak links in your own understanding

  • Rework every item you flagged as uncertain during weeks 1 through 7 into a written, correct explanation

  • Complete a timed full-length practice set and analyze errors by domain rather than by question

  • Rehearse explaining model registry promotion, job orchestration, and serving trade-offs aloud in under two minutes each

Your checkpoint

A scored full-length practice attempt with an error log grouped by domain, plus a one-page final review sheet covering your five weakest concepts

Milestones

Week 2: Register a model, create a feature table, and apply a Unity Catalog grant without consulting notes; explain library scope precedence correctly out loud

Week 4: Produce a packaged pipeline artifact from a clean notebook and regenerate the same test metrics on a rerun, within a 5 percent tolerance

Week 6: Demonstrate an unattended scheduled multi-task job whose run history shows correct task ordering and logged output metrics

Week 7: Score a live serving endpoint and complete a batch inference write to Delta, then articulate the rollback procedure from memory

Week 8: Achieve at least 80 percent on a timed full-length practice attempt with no domain scoring below 70 percent