2016-07-26 14 views
0

は、Python 3で私のコードです:予測がプロットされないのはなぜですか?ここ

from sklearn import linear_model 
import numpy as np 
obj = linear_model.LinearRegression() 
allc = np.array([[0,0],[1,1],[2,2],[3,3],[4,4],[5,5],[6,6]]) 
X=allc[:,0] 
X=X.reshape(-1, 1) 
Y=X.reshape(X.shape[0],-1) 
obj.fit(X, Y) 
print(obj.predict(7)) 
import matplotlib.pyplot as plt 
plt.scatter(X,Y,color='black') 
plt.plot(X[0],obj.predict(7),color='black',linewidth=3) 
plt.show() 

私はこのように見えるデータをプロット:フィッティング、obj.predict後 enter image description here を(7)に等しい[7]

私が間違って何をやっています?私は7.7ポイントがプロットされることを期待しました。

+0

期待される出力は? – DavidG

答えて

1

プロット方法は、X軸用の配列とY軸用の配列をとり、これらの配列に従っての線を描画します。これで

plt.plot(X[0],obj.predict(7),color='black',linewidth=3) 

:あなたはあなたのコードが動作するために(私はそれをテストしてきたし、それが働いていた)、このラインを切り替える

... ラインするための方法を使用してポイントを描画しようとしましたline:

plt.scatter(7,obj.predict(7),color='black',linewidth=3) 

scatterメソッドは指定したポイント(7,7)をとり、望むようにグラフに配置します。

私はこれが助けて欲しいと思っています:)

関連する問題