This content originally appeared on DEV Community and was authored by Phylis Jepchumba
From our previous article we looked at the machine learning steps. Lets now have a look at how to implement a machine learning model using Python.
The dataset used is collected from kaggle.
We will be able to predict the insurance amount for a person.
- We start by importing necessary modules as shown:
import pandas as pd
from sklearn.preprocessing import LabelEncoder
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression
from sklearn.metrics import accuracy_score
- Then import the data.
data=pd.read_csv('insurance.csv')
data
- Clean the data by removing duplicate values and transform the columns into numerical values to make the easier to work with.
label=LabelEncoder()
label.fit(data.sex.drop_duplicates())
data.sex=label.transform(data.sex)
label.fit(data.smoker.drop_duplicates())
data.smoker=label.transform(data.smoker)
label.fit(data.region.drop_duplicates())
data.region=label.transform(data.region)
data
The final dataset is as shown below;
- Using the cleaned dataset, now split it into training and test sets.
X=data.drop(['charges'], axis=1)
y=data[['charges']]
X_train, X_test, y_train, y_test = train_test_split( X, y, test_size=0.33, random_state=42)
- After splitting the model choose the suitable algorithm. In this case we will use Linear Regression since we need to predict a numerical value based on some parameters.
model=LinearRegression())
model.fit(X_train,y_train)
- Now predict the testing dataset and find how accurate your predictions are.
- Accuracy score is predicted as follows:
- parameter tuning Lets find the hyperparameters which affect various variables in the dataset.
This content originally appeared on DEV Community and was authored by Phylis Jepchumba
Print
Share
Comment
Cite
Upload
Translate
Updates
There are no updates yet.
Click the Upload button above to add an update.
APA
MLA
Phylis Jepchumba | Sciencx (2021-08-18T07:24:02+00:00) Implementing Machine Learning steps using Regression Model.. Retrieved from https://www.scien.cx/2021/08/18/implementing-machine-learning-steps-using-regression-model/
" » Implementing Machine Learning steps using Regression Model.." Phylis Jepchumba | Sciencx - Wednesday August 18, 2021, https://www.scien.cx/2021/08/18/implementing-machine-learning-steps-using-regression-model/
HARVARDPhylis Jepchumba | Sciencx Wednesday August 18, 2021 » Implementing Machine Learning steps using Regression Model.., viewed ,<https://www.scien.cx/2021/08/18/implementing-machine-learning-steps-using-regression-model/>
VANCOUVERPhylis Jepchumba | Sciencx - » Implementing Machine Learning steps using Regression Model.. [Internet]. [Accessed ]. Available from: https://www.scien.cx/2021/08/18/implementing-machine-learning-steps-using-regression-model/
CHICAGO" » Implementing Machine Learning steps using Regression Model.." Phylis Jepchumba | Sciencx - Accessed . https://www.scien.cx/2021/08/18/implementing-machine-learning-steps-using-regression-model/
IEEE" » Implementing Machine Learning steps using Regression Model.." Phylis Jepchumba | Sciencx [Online]. Available: https://www.scien.cx/2021/08/18/implementing-machine-learning-steps-using-regression-model/. [Accessed: ]
rf:citation » Implementing Machine Learning steps using Regression Model. | Phylis Jepchumba | Sciencx | https://www.scien.cx/2021/08/18/implementing-machine-learning-steps-using-regression-model/ |
Please log in to upload a file.
There are no updates yet.
Click the Upload button above to add an update.