Seltsames Diagramm in der linearen Regression mit pyplot

Question

Ich zeichne ein lineares Regressionsmodell mit Pyplot. Unten ist mein Code.

plt.scatter(X_train, y_train, color ='red') 
plt.show() 

Wenn ich den obigen Code plotten verwendet wird, wird die Handlung wie folgt:

I aufgetragen dann die Liniendiagramm unter Verwendung des Codes:

plt.plot(X_train, regressor.predict(X_train), color = 'blue') 
plt.show() 

Es zeigt eine Linie wie erwartet.

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

Bitte lassen Sie mich wissen, wenn ich extra zu tun haben:

Aber wenn ich versuche zusammen, beide zu zeichnen, der Graph wird immer wie unten gezeigt verkorkste Kodierung, um einen linearen Regressionsgraphen richtig zu zeichnen.

Answer 1

Pyplot verbindet die Punkte zwischen den Punkten in der Reihenfolge ihres Auftretens in X_train, aber es ist normalerweise nichts über die Reihenfolge bekannt. Es wird selten sortiert. Sie müssen Ihr Array vor dem Plotten sortieren.

sorted_indices = numpy.argsort(X_train) 
sorted_X = X_train[sorted_indices] 
plt.plot(sorted_X, regressor.predict(sorted_X), color = 'blue')