跟着github上的Avik-Jain学习机器学习:
https://github.com/Avik-Jain/100-Days-Of-ML-Code
Day4-6:逻辑回归模型及其数据可视化
1.读取数据集:
根据消费者的年龄和收入,预测其是否会在4s点买车
"""
File: logisticsregression.py
"""
import pandas as pd
import matplotlib.pyplot as plt
import numpy as np
"""read the dataset
dataset: give some information about agender, age, salary
to predict if he will purchase suv"""
data = pd.read_csv('G:/MachineLearningDailyStudy-100/Day4-logisticsregression/Social_Network_Ads.csv')
#数据的第三列是年龄,第四列是收入,第五列是是否买车
x = data.iloc[:, [2, 3]].values
y = data.iloc[:, 4]
2.数