Important Update
The Guide Feature will be discontinued after December 15th, 2023. Until then, you can continue to access and refer to the existing guides.
Author avatar

Deepika Singh

Analytics Solution with Databricks

Deepika Singh

  • Jun 5, 2020
  • 9 Min read
  • Jun 5, 2020
  • 9 Min read
Data Analytics
Machine Learning


Databricks is the unified analytics solution powered by Apache Spark, which simplifies data science with a powerful, collaborative, and fully managed machine learning platform. The major analytics solution consists of the following:

  1. Collaborative data science: Simplify and accelerate data science by providing a collaborative environment for data science and machine learning models.
  1. Reliable data engineering: Large-scale data processing for batch and streaming workloads.
  1. Production machine learning: Standardize machine learning life-cycles from experimentation to production.

In this guide, you will learn how to perform machine learning using notebooks in Databricks. The following sections will guide you through five steps to build a machine learning model with Databricks.

Step One: Login to Databricks

The first step is to go to this link and click Try Databricks on the top right corner of the page.


Once you provide the details, it will take you to the following page.


You can select cloud platforms like Azure or AWS. This guide will use the community edition of Databricks. Click on the Get Started tab under the Community Edition option and complete the signup procedures. This will get your account ready, after which you can login into the account with the login credentials.


Step Two: Importing Data

After logging into your account, you will see the following Databricks page.


Click on Import and Explore Data to import the data. You will be uploading the data file Data_db.csv from the local system, and once it is successful, the following output will be displayed.


In the above output, you can see the path of the file, "File uploaded to /FileStore/tables/Data_db.csv". You will use this link later.


This guide will use a fictitious dataset of loan applicants containing 3000 observations and seven variables, as described below:

  1. Income: Annual income of the applicant (in USD)

  2. Loan_amount: Loan amount (in USD) for which the application was submitted

  3. Credit_rating: Whether the applicant's credit score is good (1) or not (0)

  4. Loan_approval: Whether the loan application was approved (1) or not (0). This is the target variable.

  5. Age: The applicant's age in years

  6. Outstanding_debt: The current outstanding debt (in USD) of the applicant from the previous loans.

  7. Interest_rate: The rate of interest being charged by the bank to the applicant.

Step Three: Create Cluster

Again, go back to the Databricks workspace and click on New Cluster under Common Tasks.


This will open the window where you can name the cluster and keep the remaining options to default. Click Create Cluster to create the cluster named "azuredbcluster2".


Once the cluster is created, the following window is displayed. The green circle ahead of the cluster name indicates that the cluster has been successfully created.


Step Four: Launch Notebook

Once again, go back to the welcome page and click on New Notebook under Common Tasks.


The following pop-up will open, and you can fill the preferred input. In this case, you will name the notebook "Databricks ML".


Once the notebook is created, it will display the following output.


Step Five: Build Machine Learning Model

The first step is to load the data, which can be done using the code below.

1import pandas as pd
2df = pd.read_csv("/dbfs/FileStore/tables/Data_db.csv", header='infer') 

The notebook view is displayed below along with the output.


The next step is to load the other required libraries and modules.

1# Import other required libraries
2import sklearn
3import numpy as np
5# Import necessary modules
6from sklearn.linear_model import LogisticRegression 
7from sklearn.model_selection import train_test_split  
8from sklearn.metrics import confusion_matrix, classification_report

Also, create the train and test arrays required to build and evaluate the machine learning model.

1# Create arrays for the features and the response variable
2y = df['Loan_approval'].values
3X = df.drop('Loan_approval', axis=1).values 
5# Create training and test sets
6X_train, X_test, y_train, y_test = train_test_split(X, y, test_size = 0.4, random_state=42) 
7X_train.shape, X_test.shape

The notebook view is displayed below along with the output.


From the above output, you can see that there are 1800 observations in train data and 1200 observations in test data.

Next, create the logistic regression classifier, logreg, and fit the classifier on the train data.

1# Create the classifier: logreg
2logreg = LogisticRegression()
4# Fit the classifier to the training data, y_train)

The above code will generate the output displayed in the notebook view below.


You have trained the model and the next step is to predict on test data and print the evaluation metrics. This is done with the code below.

1# Predict the labels of the test set: y_pred
2y_pred = logreg.predict(X_test)
4# Compute and print the confusion matrix and classification report
5print(confusion_matrix(y_test, y_pred))
6print(classification_report(y_test, y_pred))

The above code will generate the output displayed in the notebook view below.


The output above shows that the model accuracy is 67.25% while the sensitivity is 79%. The model can be further improved by doing cross-validation, features analysis, and feature engineering and, of course, by trying out more advanced machine learning algorithms. To learn more about these and other techniques with Python, please refer to the links given at the end of the guide.


In this guide, you learned about the popular unified analytics platform, Databricks. You also learned how to build and evaluate a machine learning model using the Databricks notebook.

To learn more about data science and machine learning using Python, please refer to the following guides.