Scribd Logo
Upload

Welcome to Scribd!

  • Exp 9 DM

    Uploaded by

    Sayan Satpati
    10 views3 pages
    The document describes an experiment performing simple linear regression using R programming to predict salary based on years of experience. It imports a salary dataset, splits it into training and test sets, fits a linear regression model to the training set, predicts results on the test set, and visually plots the regression line against the actual values for both sets.

    Original Description:

    Original Title

    exp_9_DM (1)

    Copyright

    © © All Rights Reserved

    Available Formats

    PDF, TXT or read online from Scribd

    Did you find this document useful?

    Is this content inappropriate?

    Report this Document
    The document describes an experiment performing simple linear regression using R programming to predict salary based on years of experience. It imports a salary dataset, splits it into training and test sets, fits a linear regression model to the training set, predicts results on the test set, and visually plots the regression line against the actual values for both sets.

    Copyright:

    © All Rights Reserved
    Download as PDF, TXT or read online from Scribd
    Flag for inappropriate content
    10 views3 pages

    Exp 9 DM

    Uploaded by

    Sayan Satpati
    The document describes an experiment performing simple linear regression using R programming to predict salary based on years of experience. It imports a salary dataset, splits it into training and test sets, fits a linear regression model to the training set, predicts results on the test set, and visually plots the regression line against the actual values for both sets.

    Copyright:

    © All Rights Reserved
    Download as PDF, TXT or read online from Scribd
    Flag for inappropriate content
    of 3

    Experiment:- 9

    Student Name: UID:


    Branch: Section/Group:
    Semester: Subject Code:
    Subject Name:

    Aim/Overview of the practical:

     To perform simple linear regression using R programming.

    Apparatus/Simulator used:
     R- Studio
     R-language
     Dataset Salary, caTools

    Code :

    # Importing the dataset


    dataset = read.csv('salary.csv')

    # Splitting the dataset into the


    # Training set and Test set
    install.packages('caTools')
    library(caTools)
    split = sample.split(dataset$Salary, SplitRatio = 0.7)
    trainingset = subset(dataset, split == TRUE)
    testset = subset(dataset, split == FALSE)

    # Fitting Simple Linear Regression to the Training set


    lm.r= lm(formula = Salary ~ YearsExperience,
    data = trainingset)
    coef(lm.r)

    # Predicting the Test set results


    ypred = predict(lm.r, newdata = testset)

    install.packages("ggplot2")
    library(ggplot2)

    # Visualising the Training set results


    ggplot() + geom_point(aes(x = trainingset$YearsExperience,
    y = trainingset$Salary), colour = 'red') +
    geom_line(aes(x = trainingset$YearsExperience,
    y = predict(lm.r, newdata = trainingset)), colour = 'blue') +
    ggtitle('Salary vs Experience (Training set)') +
    xlab('Years of experience') +
    ylab('Salary')

    # Visualising the Test set results


    ggplot() +
    geom_point(aes(x = testset$YearsExperience, y = testset$Salary),
    colour = 'red') +
    geom_line(aes(x = trainingset$YearsExperience,
    y = predict(lm.r, newdata = trainingset)),
    colour = 'blue') +
    ggtitle('Salary vs Experience (Test set)') +
    xlab('Years of experience') +
    ylab('Salary')

    # Visualizing the Testing set results


    ggplot() + geom_point(aes(x = trainingset$YearsExperience,
    y = trainingset$Salary), colour = 'red') +
    geom_line(aes(x = trainingset$YearsExperience,
    y = predict(lm.r, newdata = trainingset)), colour = 'blue') +

    ggtitle('Salary vs Experience (Training set)') +


    xlab('Years of experience') +
    ylab('Salary')

    # Visualising the Test set results


    ggplot() +
    geom_point(aes(x = testset$YearsExperience, y = testset$Salary),
    colour = 'red') +
    geom_line(aes(x = trainingset$YearsExperience,
    y = predict(lm.r, newdata = trainingset)),
    colour = 'blue') +
    ggtitle('Salary vs Experience (Test set)') +
    xlab('Years of experience') +
    ylab('Salary')

    OUTPUT:

    You might also like