Python program: Prediction with Logistic Regression

Social Work 17
June 20, 2022
Training manual Devan
June 20, 2022
Show all

Python program: Prediction with Logistic Regression

 

The file Invistico_Airline_LR.csv contains information from an airline using the alias Invistico Airline on customer satisfaction, as well as details on each customer. The columns of interest are Gender, Age, Class, Arrival_Delay_in_Minutes, and satisfaction.

  • Read the file Invistico_Airline_LR.csv into a data frame.
  • Obtain user defined values female, age, economy, and delay.
  • Re-code the categorical variables Gender, Class, and satisfaction into dummy variables.
  • Create a new data frame X from the predictor variables Gender_female, Age, Class_Eco, and Arrival_Delay_in_Minutes, in that order.
  • Create a response variable Y from the dummy variable satisfaction_satisfied.
  • Perform logistic regression on X and Y.
  • Use the user defined values to predict the probability that a customer with those values is satisfied.

Ex: If the input is 1 34 0 10 the ouput is:

[0.62343979]

Leave a Reply

Your email address will not be published. Required fields are marked *