book

The Data Lakehouse Revolution: Harnessing the Power of Databricks for Generative AI and Machine Learning

Name: The Data Lakehouse Revolution: Harnessing the Power of Databricks for Generative AI and Machine Learning
Author: Rajaniesh Kaushikk
ISBN: 9798868817212

by Rajaniesh Kaushikk

October 2025

Intermediate to advanced

469 pages

10h 11m

English

Apress

Read now

Unlock full access

The Data Lakehouse Revolution
Foreword
Introduction
Who This Book Is ForWhat This Book CoversHow to Use This Book
Table of Contents
About the Author
About the Technical Reviewer
1. Getting Started with Databricks
Overview of DatabricksA Unified Data Life Cycle PlatformFlexibility and ScalabilityCollaboration CapabilitiesCloud-Native and Open by DesignSupport for All Data PersonasReal-Time Analytics and AI-Driven DiscoveriesGovernance and SecurityData Ecosystem IntegrationBusiness Use Cases and Industry ApplicationsDatabricks Unique PropositionKey Features and Benefits of DatabricksLakehouse Architecture: Bridging Data Warehouses and Data LakesKey Characteristics of Lakehouse ArchitectureUnified and Versatile Data StorageBreaking Down Data SilosUnified Solution with the Lakehouse ArchitectureFlexibility Across All Data FormatsElimination of Data DuplicationHigh-Performance QueriesDelta Lake for Transactional CapabilitiesData Versioning and Time TravelCollaborative NotebooksKey Features of Collaborative NotebooksReal-Time Multiuser EditingEnhanced Communication ToolsMulti-language SupportManaged Compute ClustersKey Features of Managed Compute ClustersAutoscaling for Dynamic WorkloadsCost Efficiency Through Resource OptimizationPreconfigured Environments for SimplicityStreamlined Management and MonitoringAI and Machine Learning IntegrationKey Features of AI and Machine Learning IntegrationMLflow IntegrationPreconfigured LibrariesSimplified DeploymentSecurity and Governance: Safeguarding Data with ConfidenceKey Features of Security and GovernanceRole-Based Access Control (RBAC)Unity CatalogCore Functions of Unity CatalogData EncryptionCloud IntegrationsKey Features of Cloud IntegrationNative Support for Cloud StorageScalability Across Cloud RegionsCollaboration with Cloud-Native ServicesElastic Compute Resources: Adapting to Dynamic WorkloadsDelta Sharing for Data CollaborationKey Features of Delta SharingReal-Time Access to Live DataFine-Grained Access ControlOpen Standard for InteroperabilityScalable for Growing DemandsBuilt-In Security and ComplianceDelta Sharing in ActionSupport for Diverse WorkloadsComprehensive ETL PipelinesIntegrated Machine Learning CapabilitiesSQL for Business IntelligenceUnified Batch and Streaming Data ProcessingAdvanced AI WorkflowsUnified Workload Management in ActionCost and Performance OptimizationKey Features of Databricks for Cost and Performance OptimizationPhoton Engine for Accelerated QueriesAutoscaling Compute ClustersAdaptive Query Execution (AQE)Cloud-Native Cost-Saving FeaturesDelta Lake for Optimized StorageCost and Performance Optimization at ScaleMonitoring and Logging capabilitiesKey Features of Databricks Monitoring and LoggingReal-Time Job DashboardsCluster Performance MetricsDetailed Logs for Debugging and ComplianceAlerts and NotificationsAudit Trails and ComplianceHistorical Metrics for Trend AnalysisMonitoring and Logging in ActionKey Components of DatabricksNotebooks: Your Interactive Development EnvironmentDatabricks ClustersDatabricks JobsDatabricks WorkflowsNavigating the Interface: Step-by-StepCreating the Databricks AccountStep 1: Choosing the Cloud Provider: Finding the Right Fit for Your Databricks EnvironmentFactors to Consider When Choosing a Cloud ProviderStep 2: Registering Your Databricks AccountStep 3: Configuring Your WorkspaceNaming and Provisioning Your WorkspaceStep 4: Configuring Compute ClustersStep 5: Managing User Access and RolesStep 6: Integrating Data SourcesWorkspace Deployment in ActionTips and Best Practices for Efficient Navigation in DatabricksOrganize Your Workspace for ClarityMaster Keyboard ShortcutsOptimize Cluster UsageLeverage Search and Filtering ToolsCollaborate EffectivelyMonitor Jobs and ClustersUse Workflows to Automate Repetitive TasksKeep Your Environment CleanDatabricks Navigation in ActionHands-On Lab: Setting Up and Navigating DatabricksLogging into DatabricksSetting Up Your WorkspaceConfiguring Compute ClustersWorking with NotebooksSummary
2. Introduction to Machine Learning and Data Lakehouses
Overview of Machine LearningWhy Machine Learning MattersKey Concepts in Machine LearningData: The Fuel for LearningFeatures: The Building Blocks of PredictionsModel: The Brain of the OperationTraining and Testing: The Learning ProcessEvaluation Metrics: The Report CardOverfitting and Underfitting: Finding the BalanceTypes of Machine LearningSupervised Learning: Learning with a TeacherUnsupervised Learning: Finding Hidden PatternsReinforcement Learning: Learning Through RewardsComparing the Three Types of Machine LearningReal-World Examples and Use CasesHands-On Lab: Practical Exercises on Basic ML ConceptsExercise 1: Supervised Learning—Predicting Housing PricesLoad the DatasetSplit Data into Features and TargetSplit the Data into Training and Testing SetsBuild and Train the ModelEvaluate the ModelVisualize ResultsComparing Polynomial Regression and Linear RegressionComparing Linear vs. Polynomial vs. Random Forest RegressionExercise 2: Unsupervised Learning—Customer SegmentationLoad the DatasetPrepare Data for ClusteringPlot the Data to Understand ItApply K-Means ClusteringVisualize the ClustersInterpret the ClustersDetermine the Optimal Number of Clusters: Elbow MethodAdding Age As the Third Parameter in the Clustering and How to Interpret the ClusteringUse the Silhouette Score to Evaluate the Quality of ClusteringSummary
3. Data Preparation and Management
Introduction to Data IngestionBatch IngestionHow to Implement Batch IngestionStreaming IngestionHow to Implement Streaming IngestionBatch vs. Streaming IngestionStandard Data Ingestion Tools and MethodsData Cleaning and TransformationTechniques for Data Cleaning and PreprocessingHandling Missing ValuesForward/Backward Fill (Time-Series Data)Row/Column RemovalRemoving DuplicatesStandardizing Data TypesOutlier Detection and HandlingData Normalization and ScalingSummary of Data Cleaning and Transformation TechniquesTools and Libraries for Data TransformationOpen Source LibrariesData Integration PlatformsCloud-Native SolutionsSpecialized Transformation ToolsMachine Learning-Driven Transformation ToolsGovernance and Catalog IntegrationBest Practices for Data TransformationManaging Data with Unity CatalogKey Problems Unity Catalog SolvesHow Unity Catalog WorksBest Practices for Data Organization and GovernanceHands-On LabsExercise 1: Batch Processing with DatabricksStep 1: Create Directories in DBFSStep 2: Write Sample Data for Store A to CSVStep 3: Write Sample Data for Store B to JSONStep 4: Load Data into Spark DataFrames and Save As Delta TablesStep 5: Query the Delta TablesStep 6: Query the Delta TablesStep 7: Consolidate Data into a New TableStep 8: Verify Consolidated DataExercise 2: Streaming Data Processing in DatabricksStep 1: Simulate Real-Time Data StreamsStep 2: Define the Schema and Load Streaming DataStep 3: Process and Save Streaming Data to a Delta TableStep 4: Register and Query the Streaming Delta TableExercise 3: Handling and Cleaning Batch Data in DatabricksStep 1: Create a Directory to Host the Demo FilesStep 2: Prepare Output DirectoryStep 3: Generate Sample DataStep 4: Check If the File Is Saved to the DirectoryStep 5: Load Data into a Spark DataFrameStep 6: Validate Data LoadingStep 7: Handle Missing ValuesStep 8: Verify Missing ValuesStep 9: Calculate the Median for the Available QuantitySold ValuesStep 10: Update Missing ValuesStep 11: Analyze the Cleaned DataExercise 4: Managing Data with Unity Catalog in DatabricksStep 1: Load and Standardize DataStep 2: Create a CatalogStep 3: Create a SchemaStep 4: Save Data to Unity CatalogStep 5: Query the Standardized DataStep 6: Verify Table OrganizationStep 7: Create User GroupsStep 8: Assign PermissionsStep 9: Validate PermissionsSummary
4. Building Machine Learning Models
Fundamentals of Machine Learning ModelsIntroduction to MLflowKey Features and ComponentsMLflow TrackingMLflow ProjectsMLflow ModelsMLflow RegistryStep-by-Step Guide to Creating an MLflow Project in Azure DatabricksCreate a New Databricks NotebookSet Up MLflow Experiment TrackingLoad and Prepare DataTrain and Log a Model Using MLflowView MLflow Experiment Tracking UIRegister the Model in the MLflow RegistryLoad and Use the Registered Model for InferenceChoosing the Right Algorithm for Model TrainingWhy Is Choosing the Right Algorithm Important?Factors to Consider When Choosing an Algorithm and Why It Is ImportantType of ProblemSize and Quality of DataInterpretability vs. AccuracyComputational RequirementsData TypeScenario: Choosing the Right Algorithm for Customer Churn PredictionIdentifying the Type of ProblemConsidering the Data CharacteristicsBalancing Interpretability vs. AccuracyComputational RequirementsChoosing the Right Algorithm Based on Data TypeModel Training and Hyperparameter TuningWhat Are Hyperparameters?What Is Hyperparameter Tuning?Why Is Hyperparameter Tuning Important?Evaluating Model PerformanceKey Metrics for Evaluating Model PerformanceAccuracy: How Often the Model Is CorrectPrecision: How Many of the Positive Predictions Were CorrectRecall (Sensitivity): How Many Actual Positive Cases Were DetectedF1-Score: Balancing Precision and RecallUnderstanding Errors: False Positives vs. False NegativesChoosing the Right MetricEnd-to-End Example: Evaluating Model Performance and Logging with MLflowTrain a Classification ModelMake Predictions and Compute Evaluation MetricsLog Metrics with MLflowInterpret the MetricsViewing MLflow Logs in DatabricksExperiment Tracking with MLflowWhy Experiment Tracking Is ImportantRecording and Managing ExperimentsLogging Parameters and MetricsManaging ArtifactsVisualizing Experiment ResultsHands-On LabsLab 1: Hyperparameter Tuning with MLflow and HyperoptLab 2: Training a Random Forest Model with MLflowLab 3: Loan Default Prediction LabSetup Databricks File System (DBFS)Create and Upload Loan DataLoad and Display DataLoad Data into PySparkSet Up MLflow ExperimentTrain the Loan Default Prediction ModelRegister Model in MLflowLoad and Make PredictionsVisualizing Accuracy Across Different RunsInterpreting the Accuracy PlotKey ObservationsRecommended Next StepsUnderstanding the Metrics: Precision, Recall, and F1-ScoreWhat Does Precision: 0.00 Mean?What Does Recall: 0.00 Mean?What Does F1-Score: 0.00 Mean?Why Are All Metrics 0.00? Common ReasonsHow to Fix It: RecommendationsInterpreting the Confusion MatrixStructure of the Confusion MatrixExplanation of Each TermKey Insights from the Confusion MatrixHow to Interpret the MatrixNext Steps to Improve PerformanceInterpreting Validation Accuracy: 1.00What Does Validation Accuracy: 1.00 Mean?Is a Validation Accuracy of 1.00 Too Good to Be True?How to Confirm If Validation Accuracy Is GenuineRecommended Next StepsVisualizing the Experiment ResultSummary

5. AutoML and Model Optimization
Introduction to AutoMLWhat Is AutoML?Why Use AutoML?Use Cases of AutoMLHow AutoML Improves Fraud Detection?How AutoML Works?Data PreprocessingFeature EngineeringModel SelectionHyperparameter TuningEvaluation and RankingDeploymentBenefits and Limitations of AutoMLBenefits of using AutoMLChallenges and LimitationsWhy Choose Databricks AutoML?Key Advantages of Databricks AutoMLUsing Databricks AutoMLSetting Up and Running AutoML ExperimentsPreparing the DatasetLaunching AutoMLChoosing the Machine Learning TaskConfiguring the ExperimentExecuting the ExperimentReviewing the Experiment ResultsInterpreting AutoML ResultsModel LeaderboardFeature Importance AnalysisHyperparameter Tuning SummaryModel Explainability (SHAP Values)Performance MetricsExperiment Tracking with MLflowModel Optimization TechniquesTechniques for Improving Model AccuracyFeature EngineeringHyperparameter TuningData AugmentationRemoving OutliersEnsembling ModelsPerformance Tuning and Optimization StrategiesReducing Model ComplexityOptimizing Feature SelectionEfficient Data ProcessingDeploying Models with CachingScaling InfrastructureHands-On Lab 1: AutoML Experiment for Housing Price PredictionScenarioPrerequisitesGenerating the Sample DatasetRunning AutoML in DatabricksLaunch Databricks AutoMLSelect the DatasetChoose Prediction TypeConfigure the Advanced Configuration Section in ExperimentsJoin FeaturesInterpreting AutoML ResultsReviewing the ExperimentTesting the Model After RegistrationDatabricks AutoML Python APIWhy Use the AutoML Python API?Running AutoML Using Python APIEvaluating and Deploying the Best ModelRetrieving AutoML ResultsLoading and Testing the Best ModelHands-On Lab 2: Predicting Loan Defaults Using Databricks AutoML (Python API)Generate Sample Loan Application DataRun AutoML for Loan Default PredictionEvaluate AutoML ResultsDeploy and Test the Best ModelSummary
6. Deploying Machine Learning Models
Model Deployment: What It Takes to Go LiveDeployment Strategies and ConsiderationsBatch DeploymentReal-Time DeploymentEdge DeploymentDeployment ConsiderationsModel SerializationCommon Serialization FormatsHow to Perform MLflow SerializationComputational ResourcesCloud vs. Edge ConsiderationsIntegration RequirementsKey Integration AreasCollaboration and DocumentationCompliance and GovernanceKey Governance AreasImplementing Model GovernanceCloud-Based DeploymentKey Deployment Approaches in the CloudBenefits of Cloud-Based DeploymentConsiderations and ChallengesWhen to Choose Cloud-Based DeploymentOn-Premises DeploymentTypes of On-Premises EnvironmentsAdvantages of On-Premises DeploymentChallenges of On-Premises DeploymentWhen to Choose On-Premises DeploymentHybrid DeploymentCommon Hybrid Deployment PatternsAdvantages of Hybrid DeploymentChallenges of Hybrid DeploymentWhen to Choose Hybrid DeploymentDeploying Models in DatabricksUsing MLflow for Model Deployment and the Deployment ChallengeA Real-World Example: Churn Prediction Gone WrongMLflow: A Unified Solution for Deployment ChallengesUnderstanding MLflow’s Approach to DeploymentHow Does MLflow Package a Model As a Complete Artifact?How to Log a Model with MLflow?How Does MLflow Ensure Environment Reproducibility?The MLflow Deployment PhilosophyCommon Deployment Patterns Supported by MLflowBatch InferenceReal-Time API ServingEdge DeploymentEmbedded DeploymentMLflow Model Registry: Your Model’s Home in ProductionMLflow Deployment Quick ChecklistBest Practices for Deployment in ProductionEnvironment Reproducibility and the Dependency ChallengeValidating Environments Before DeploymentAdditional Best Practices for Dependency ManagementTroubleshooting Model Deployment IssuesDependency Management ErrorsCommon SymptomsSolutionsData Schema MismatchesResource ConstraintsNetworking and Connectivity IssuesCommon SymptomsRecommended SolutionsModel Serving Performance IssuesTroubleshooting ChecklistExternal Integration PatternsREST API IntegrationWebhook IntegrationMessage Queue IntegrationMobile Application IntegrationIntegration Best PracticesSecure API AccessError Handling and ResilienceVersioningPerformance OptimizationMonitoring Integration PointsMonitoring and Maintaining Deployed ModelsSetting Up Monitoring and AlertingCreating Dashboards and AlertsDashboardsAlertsThe Monitoring Life CycleManaging the Model Life Cycle and UpdatesUnderstanding the Model Life CycleTriggers for Model UpdatesHands-On Exercise: Predict Customer Churn and Serve the Model via REST APISummary
7. Advanced Topics in Machine Learning
Overview of Explainability Techniques: SHAP, LIME, and BeyondSHAPLIMEPartial Dependence PlotsEthical Considerations in Machine LearningRecognizing and Addressing BiasEnsuring Fairness and AccountabilityRegulatory Implications and ComplianceFuture Trends in Machine LearningEmerging Technologies and ArchitecturesFuture Trends in Machine LearningEmerging Technologies and ArchitecturesReal-Time and Continual LearningML in the Age of Generative AIHands-On Lab: Explainability and Governance for Loan Default Risk PredictionCreate a Realistic Loan Default DatasetTrain a Binary Classification ModelGenerate SHAP ExplanationsLog SHAP Artifacts with MLflowRegister the Model and Enrich with Unity Catalog MetadataTrain a New Model Version and Compare InterpretabilitySummary
8. Lakehouse AI and Retrieval-Augmented Generation (RAG)
Introduction to Lakehouse AIKey Features and CapabilitiesBenefits of Lakehouse AIRetrieval-Augmented Generation (RAG): A Foundation for Enterprise LLMsWhat Is RAG and Why Does It Matter?RAG Architecture and ComponentsModel and Tooling OptionsBuilding RAG Pipelines with Lakehouse AIImplementing RAG with Delta Tables and Vector SearchImplementing Governance on RAG PipelinesPrompt Engineering and GuardrailsWhat Is Prompt Engineering?Tools for Prompt Management and TestingReal-World Use Cases and Industry ApplicationsHealthcare: Clinical Note SummarizationFinance: Fraud Detection and Document AnalysisRetail: Intelligent Product RecommendationsLab: Building a Vector Search-Powered HR Chatbot on DatabricksStep 1: Install Required LibrariesStep 2: Restart the Python KernelStep 3: Create Catalog and Schema Using PySparkStep 4: Load HR Policy Sample Data into a Delta TableStep 5: Verify HR Policy Data in SQLStep 6: Create and Validate Vector Search EndpointStep 7: Create and Sync Vector Search IndexStep 8: Perform Semantic Search on the Vector IndexStep 9: Define LangChain Configuration for the HR ChatbotStep 10: Constructing the Retriever Pipeline with LangChain and Vector SearchStep 11: Building the LLM Chain to Generate HR AnswersPrompt Template SetupModel InitializationTesting the ChainDisplaying the AnswerStep 12: Enable MLflow Autologging for LangChainStep 13: Building the Full RAG Chatbot Pipeline with LangChain, Vector Search, and Databricks LLMChain ConfigurationRetriever ComponentContext FormatterPrompt Template and LLMUser Query ExtractionFull LangChain RAG ChainRun an Example QueryDisplay the Answer☑ SummarySummary
9. Conclusion and Next Steps
Recap of Key ConceptsGetting Started with DatabricksIntroduction to Machine Learning and LakehousesData Preparation and ManagementBuilding ML Models with MLflowAutoML and Model OptimizationDeploying ModelsResponsible AI and GovernanceLakehouse AI and Retrieval-Augmented Generation (RAG)Resources for Further LearningOfficial Databricks ResourcesBooks and Technical ReferencesOnline Courses and CertificationsCommunities and ForumsGitHub Repositories and TemplatesNext Steps in Your ML JourneyBuild and Operationalize Your Own ProjectsJoin or Lead a Data Project TeamContribute to Open Source and CommunitySeek Feedback and Reflect on ImpactPlan Your Career GrowthSummaryKey TakeawaysLooking AheadFinal WordsCall to Action
Index

Content preview from The Data Lakehouse Revolution: Harnessing the Power of Databricks for Generative AI and Machine Learning

R. KaushikkThe Data Lakehouse Revolutionhttps://doi.org/10.1007/979-8-8688-1721-2_4

4. Building Machine Learning Models

Rajaniesh Kaushikk¹

Machine learning models are at the core of artificial intelligence and data-driven decision-making, enabling systems to learn from data and make informed predictions. However, the process of building, training, and tracking these models can be complex, requiring robust tools and frameworks to ensure efficiency and reproducibility.

One such powerful tool is MLflow, an open source platform designed to manage the entire machine learning life cycle. MLflow helps data scientists and engineers ...

Become an O’Reilly member and get unlimited access to this title plus top books and audiobooks from O’Reilly and nearly 200 top publishers, thousands of courses curated by job role, 150+ live events each month,
and much more.

Read now

Unlock full access

More than 5,000 organizations count on O’Reilly

O’Reilly covers everything we've got, with content to help us build a world-class technology community, upgrade the capabilities and competencies of our teams, and improve overall team performance as well as their engagement.

Julian F.

Head of Cybersecurity

I wanted to learn C and C++, but it didn't click for me until I picked up an O'Reilly book. When I went on the O’Reilly platform, I was astonished to find all the books there, plus live events and sandboxes so you could play around with the technology.

Addison B.

Field Engineer

I’ve been on the O’Reilly platform for more than eight years. I use a couple of learning platforms, but I'm on O'Reilly more than anybody else. When you're there, you start learning. I'm never disappointed.

Amir M.

Data Platform Tech Lead

I'm always learning. So when I got on to O'Reilly, I was like a kid in a candy store. There are playlists. There are answers. There's on-demand training. It's worth its weight in gold, in terms of what it allows me to do.

Mark W.

Embedded Software Engineer

ML and Generative AI in the Data Lakehouse

Publisher Resources

ISBN: 9798868817212Purchase Link Publisher Website

Cloud Computing

Data Engineering

Data Science

AI & ML

Programming Languages

Software Architecture

IT/Ops

Security

Design

Business

Soft Skills

The Data Lakehouse Revolution: Harnessing the Power of Databricks for Generative AI and Machine Learning

by Rajaniesh Kaushikk

4. Building Machine Learning Models

Become an O’Reilly member and get unlimited access to this title plus top books and audiobooks from O’Reilly and nearly 200 top publishers, thousands of courses curated by job role, 150+ live events each month,
and much more.