Building Your First Insurance Cost Prediction Model with Simple Linear Regression

Introduction & Lesson Overview

Welcome back! In the last lesson, you learned how to identify which factors most strongly influence insurance costs in the PredictHealth dataset. You explored correlations, visualized relationships, and compared group averages to spot the most important predictors of insurance charges. Now, you are ready to take the next step: building your very first cost prediction model. In this lesson, you will learn how to use simple linear regression to predict insurance charges based on a single feature — age . By the end of this lesson, you will know how to prepare your data for modeling, train a regression model, evaluate its performance, visualize the results, and use the model to make predictions for new customers. This is a key skill in data science and will help you understand how predictive models work in real-world insurance scenarios.

Understanding Simple Linear Regression

Simple linear regression is a basic but powerful tool in data science. It helps you predict a numeric outcome (like insurance charges) using just one input feature (like age). The model fits a straight line to the data, which can be described by the equation: y = intercept + slope \times x y = intercept + slope \times x Here, y is the predicted value (insurance charges), x is the input feature (age), the slope shows how much y changes for each unit increase in x, and the intercept is the value of y when x is zero. In the context of PredictHealth, you will use age as the predictor. This means you are asking, "If I know a customer's age, how much can I expect their insurance cost to be?" The regression model will find the best-fitting line through the data so you can make these predictions.

Data Preparation For Regression

Before building a prediction model, it is important to select the right variables. In previous lessons, you discovered that age is one of the features most strongly related to insurance charges. For this first regression model, you will use age as the input (feature) and charges as the output (target) you want to predict. To prepare the data, you need to create two variables: X for the feature and y for the target. In this case, X will be a DataFrame containing the age column, and y will be a Series containing the charges column. Here is how you can do this: Python X = insurance_data[['age']] # Feature: age y = insurance_data['charges'] # Target: charges X = insurance_data[['age']] # Feature: age y = insurance_data['charges'] # Target: charges This code selects the age column as the feature and the charges column as the target. Remember, you are using double square brackets for X to keep it as a DataFrame, which is required by scikit-learn . This step sets up your data for the regression model.

Join the 1M+ learners on CodeSignal

Be a part of our community of 1M+ users who develop and demonstrate their skills on CodeSignal

y = intercept + slope × x

Python

X = insurance_data[['age']]  # Feature: age
y = insurance_data['charges']  # Target: charges

Python

from sklearn.model_selection import train_test_split

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

from sklearn.model_selection import train_test_split

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

Python

from sklearn.linear_model import LinearRegression

model = LinearRegression()
model.fit(X_train, y_train)

from sklearn.linear_model import LinearRegression

model = LinearRegression()
model.fit(X_train, y_train)

y_pred = model.predict(X_test)

from sklearn.metrics import mean_squared_error, r2_score
import numpy as np

mse = mean_squared_error(y_test, y_pred)
rmse = np.sqrt(mse)
r2 = r2_score(y_test, y_pred)

print(f"Model Coefficients:")
print(f"Slope (age coefficient): {model.coef_[0]:.2f}")
print(f"Intercept: {model.intercept_:.2f}")
print("\nModel Equation:")
print(f"Insurance Cost = {model.intercept_:.2f} + {model.coef_[0]:.2f} × Age")
print("\nModel Evaluation:")
print(f"Mean Squared Error (MSE): {mse:.2f}")
print(f"Root Mean Squared Error (RMSE): {rmse:.2f}")
print(f"R-squared (R²): {r2:.2f}")

Model Coefficients:
Slope (age coefficient): 257.72
Intercept: 3165.89

Model Equation:
Insurance Cost = 3165.89 + 257.72 × Age

Model Evaluation:
Mean Squared Error (MSE): 148000000.00
Root Mean Squared Error (RMSE): 12162.29
R-squared (R²): 0.09

Python

import matplotlib.pyplot as plt

plt.figure(figsize=(10, 6))
plt.scatter(X_test, y_test, color='blue', alpha=0.5, label='Actual')
plt.plot(X_test, y_pred, color='red', linewidth=2, label='Predicted')
plt.title('Age vs. Insurance Charges: Simple Linear Regression')
plt.xlabel('Age')
plt.ylabel('Charges ($)')
plt.legend()
plt.tight_layout()
plt.show()

import matplotlib.pyplot as plt

plt.figure(figsize=(10, 6))
plt.scatter(X_test, y_test, color='blue', alpha=0.5, label='Actual')
plt.plot(X_test, y_pred, color='red', linewidth=2, label='Predicted')
plt.title('Age vs. Insurance Charges: Simple Linear Regression')
plt.xlabel('Age')
plt.ylabel('Charges ($)')
plt.legend()
plt.tight_layout()
plt.show()

Python

new_customer_age = 45
new_customer_data = pd.DataFrame({'age': [new_customer_age]})
predicted_cost = model.predict(new_customer_data)
print(f"\nPredicted insurance cost for a {new_customer_age}-year-old customer: ${predicted_cost[0]:.2f}")

new_customer_age = 45
new_customer_data = pd.DataFrame({'age': [new_customer_age]})
predicted_cost = model.predict(new_customer_data)
print(f"\nPredicted insurance cost for a {new_customer_age}-year-old customer: ${predicted_cost[0]:.2f}")

Predicted insurance cost for a 45-year-old customer: $14762.29

Predicted insurance cost for a 45-year-old customer: $14762.29

Python

from sklearn.model_selection import train_test_split

X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)

Python

from sklearn.linear_model import LinearRegression

model = LinearRegression()
model.fit(X_train, y_train)

Python

y_pred = model.predict(X_test)

y_pred = model.predict(X_test)

Python

from sklearn.metrics import mean_squared_error, r2_score
import numpy as np

mse = mean_squared_error(y_test, y_pred)
rmse = np.sqrt(mse)
r2 = r2_score(y_test, y_pred)

print(f"Model Coefficients:")
print(f"Slope (age coefficient): {model.coef_[0]:.2f}")
print(f"Intercept: {model.intercept_:.2f}")
print("\nModel Equation:")
print(f"Insurance Cost = {model.intercept_:.2f} + {model.coef_[0]:.2f} × Age")
print("\nModel Evaluation:")
print(f"Mean Squared Error (MSE): {mse:.2f}")
print(f"Root Mean Squared Error (RMSE): {rmse:.2f}")
print(f"R-squared (R²): {r2:.2f}")

from sklearn.metrics import mean_squared_error, r2_score
import numpy as np

mse = mean_squared_error(y_test, y_pred)
rmse = np.sqrt(mse)
r2 = r2_score(y_test, y_pred)

print(f"Model Coefficients:")
print(f"Slope (age coefficient): {model.coef_[0]:.2f}")
print(f"Intercept: {model.intercept_:.2f}")
print("\nModel Equation:")
print(f"Insurance Cost = {model.intercept_:.2f} + {model.coef_[0]:.2f} × Age")
print("\nModel Evaluation:")
print(f"Mean Squared Error (MSE): {mse:.2f}")
print(f"Root Mean Squared Error (RMSE): {rmse:.2f}")
print(f"R-squared (R²): {r2:.2f}")

Model Coefficients:
Slope (age coefficient): 257.72
Intercept: 3165.89

Model Equation:
Insurance Cost = 3165.89 + 257.72 × Age

Model Evaluation:
Mean Squared Error (MSE): 148000000.00
Root Mean Squared Error (RMSE): 12162.29
R-squared (R²): 0.09

Model Coefficients:
Slope (age coefficient): 257.72
Intercept: 3165.89

Model Equation:
Insurance Cost = 3165.89 + 257.72 × Age

Model Evaluation:
Mean Squared Error (MSE): 148000000.00
Root Mean Squared Error (RMSE): 12162.29
R-squared (R²): 0.09

Python

import matplotlib.pyplot as plt

plt.figure(figsize=(10, 6))
plt.scatter(X_test, y_test, color='blue', alpha=0.5, label='Actual')
plt.plot(X_test, y_pred, color='red', linewidth=2, label='Predicted')
plt.title('Age vs. Insurance Charges: Simple Linear Regression')
plt.xlabel('Age')
plt.ylabel('Charges ($)')
plt.legend()
plt.tight_layout()
plt.show()

Python

new_customer_age = 45
new_customer_data = pd.DataFrame({'age': [new_customer_age]})
predicted_cost = model.predict(new_customer_data)
print(f"\nPredicted insurance cost for a {new_customer_age}-year-old customer: ${predicted_cost[0]:.2f}")

Predicted insurance cost for a 45-year-old customer: $14762.29

Python

y_pred = model.predict(X_test)

Python

from sklearn.metrics import mean_squared_error, r2_score
import numpy as np

mse = mean_squared_error(y_test, y_pred)
rmse = np.sqrt(mse)
r2 = r2_score(y_test, y_pred)

print(f"Model Coefficients:")
print(f"Slope (age coefficient): {model.coef_[0]:.2f}")
print(f"Intercept: {model.intercept_:.2f}")
print("\nModel Equation:")
print(f"Insurance Cost = {model.intercept_:.2f} + {model.coef_[0]:.2f} × Age")
print("\nModel Evaluation:")
print(f"Mean Squared Error (MSE): {mse:.2f}")
print(f"Root Mean Squared Error (RMSE): {rmse:.2f}")
print(f"R-squared (R²): {r2:.2f}")

Model Coefficients:
Slope (age coefficient): 257.72
Intercept: 3165.89

Model Equation:
Insurance Cost = 3165.89 + 257.72 × Age

Model Evaluation:
Mean Squared Error (MSE): 148000000.00
Root Mean Squared Error (RMSE): 12162.29
R-squared (R²): 0.09

Introduction & Lesson Overview

Understanding Simple Linear Regression

Data Preparation For Regression

Join the 1M+ learners on CodeSignal

Splitting The Data & Model Training

Evaluating The Model Performance

Visualizing The Regression Line

Making Predictions For New Data

Summary & Preparation For Practice Exercises