美文网首页
day2-简单线性回归(Simple Linear Regres

day2-简单线性回归(Simple Linear Regres

作者: 卡拉肖克璠 | 来源:发表于2019-01-08 23:24 被阅读0次
    Day 2.jpg

    第二天:简单线性回归
    step1:数据初处理。(同第一天的内容:导入库,导入数据集、处理丢失数据、分隔数据集、特征缩放。) step2:用训练集匹配简单的线性模型(使用sklearn.linear_modle的LinearRegression类,然后在使用LinearRegression类的regressor对象,再用LinearRegression类中的fit()方法来训练regressor对象) step3:预测结果(通过测试集来预测结果,把结果保存在向量Y_pred) step4:可视化(使用matplotlib.pyplot库来制作散点图)

    代码:

    step2 使用线性模型来训练

    from sklearn.linear_model import LinearRegression
    regressor = LinearRegression()
    regressor = regressor.fit(X_train,Y_train)

    step3 预测结果

    Y_pred = regressor.predict(X_test)

    step4 可视化

    plt.scatter(X_train,Y_train,color = 'red')
    plt.plot(X_train,regressor.predict(X_train),color='blue')

    plt.scatter(X_test,Y_test,color='red')
    plt.plot(X_test,regressor.predict(X_test),color='blue')

    plt.show() #让图像显示出来

    相关文章

      网友评论

          本文标题:day2-简单线性回归(Simple Linear Regres

          本文链接:https://www.haomeiwen.com/subject/dljirqtx.html