首页
学习
活动
专区
工具
TVP
发布
社区首页 >问答首页 >无法在Python中使用散点图正确绘制回归线

无法在Python中使用散点图正确绘制回归线
EN

Stack Overflow用户
提问于 2018-05-31 22:20:05
回答 1查看 1.7K关注 0票数 1

我正在学习一门关于EdX的课程,内容是在数据科学中使用Python编程。当使用给定的函数来绘制我的线性回归模型的结果时,该图看起来非常不稳定,所有的散点都聚集在底部,而回归线则在顶部。

我不确定是定义的函数drawLine不正确,还是我的建模过程中有其他问题。

下面是已定义的函数

代码语言:javascript
复制
def drawLine(model, X_test, y_test, title, R2):
    fig = plt.figure()
    ax = fig.add_subplot(111)
    ax.scatter(X_test, y_test, c='g', marker='o')
    ax.plot(X_test, model.predict(X_test), color='orange', linewidth=1, alpha=0.7)

    title += " R2: " + str(R2)
    ax.set_title(title)
    print(title)
    print("Intercept(s): ", model.intercept_)

    plt.show()

下面是我写的代码

代码语言:javascript
复制
import pandas as pd
import numpy as np
import matplotlib
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from sklearn import linear_model
from sklearn.model_selection import train_test_split

matplotlib.style.use('ggplot') # Look Pretty

# Reading in data
X = pd.read_csv('Datasets/College.csv', index_col=0)

# Wrangling data
X.Private = X.Private.map({'Yes':1, 'No':0})

# Splitting data
roomBoard = X[['Room.Board']]
accStudent = X[['Accept']]
X_train, X_test, y_train, y_test = train_test_split(roomBoard, accStudent, test_size=0.3, random_state=7)

# Training model
model = linear_model.LinearRegression()
model.fit(X_train, y_train)
score = model.score(X_test, y_test)

# Visualise results
drawLine(model, X_test, y_test, "Accept(Room&Board)", score)

我使用的数据可以在here中找到

谢谢您抽时间见我。

任何帮助或建议都是非常感谢的。

EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/50626539

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档