databricks logo
Data Engineering

Real file from a public repo · synced Aug 26, 2026

databricks-ml-training

Train ML models on Databricks. Use for: classification/regression/deep-learning (XGBoost, scikit-learn, LightGBM, PyTorch) with Optuna, @prod/@challenger aliases, batch scoring (spark_udf for plain models, fe.score_batch for feature-store-backed), custom PyFunc, custom ResponsesAgent (LangGraph + UC Function/Vector Search); UC feature tables + FeatureLookup + point-in-time joins + Lakebase online store; declarative Feature Views (create_feature, DeltaTableSource, RollingWindow/SlidingWindow/TumblingWindow, materialize_features, streaming Kafka features). NOT for: endpoint ops (databricks-model-serving), MLflow evaluation (databricks-mlflow-evaluation).

Works with

Claude CodeCursorCodex CLIGitHub CopilotGemini CLI

Install in three steps

  1. 1

    Download the ZIP

    One file per agent, plus INSTALL.md with the exact paths below.

  2. 2

    Claude Code: copy SKILL.md into your skills folder

    .claude/skills/databricks-ml-training/SKILL.md    # this project
    ~/.claude/skills/databricks-ml-training/SKILL.md   # every project
  3. 3

    Other agents: use the matching file

    skill.mdc goes in Cursor’s .cursor/rules/. AGENTS.md works as GitHub Copilot instructions, a legacy .cursorrules, or a Codex / Gemini CLI instructions file.

More Data Engineering skills

All Data Engineering skills

Check your AI visibility

One URL in, a 0–100 score and the exact fixes out.

RUN THE CHECK

Browse all the tools

15 tools across six categories
13 of them never send your data anywhere

Free · No signup · No trial clock

SEE THE DIRECTORY