Как предсказать новый вывод из построения модели linearRegression с использованием sklean? - PullRequest
0 голосов
/ 24 сентября 2018

Я следовал учебному пособию по линейной регрессии с помощью scikitlearn. Код работает отлично, и теперь я хочу предсказать новый вывод, предоставив новый вход. Я использовал набор показателей ученика и час обучения.Вот код:

import  matplotlib.pyplot as plt
import pandas as pd
from sklearn.model_selection import train_test_split
from sklearn.linear_model import LinearRegression

data=pd.read_csv("/home/crpsm/Pycharm/DataSet/student_scores.csv")

data.plot(x="Hours",y="Scores",style="o")
plt.title("Score-Hour")
plt.xlabel('Hours ')
plt.ylabel('Percentage ')


x=data.iloc[:,:-1]
y=data.iloc[:,1]

 x_train,x_test,y_train,y_test=train_test_split(x,y,train_size=0.55,random_state=5)

regression_model=LinearRegression()
regression_model.fit(x_train,y_train)

print(regression_model.coef_)
print(regression_model.intercept_)
regression_model.predict(X_test)

1 Ответ

0 голосов
/ 24 сентября 2018

prediction = regression_model.predict(X)

Пожалуйста, прочитайте документы: http://scikit -learn.org / stable / modules / generate / sklearn.linear_model.LinearRegression.html

...