线性回归模型

  1. plt.clf()
  2. plt.scatter(ages_train, net_worths_train, color="b", label="train data")
  3. plt.scatter(ages_test, net_worths_test, color="r", label="test data")
  4. plt.plot(ages_test, reg.predict(ages_test), color="black")
  5. plt.legend(loc=2)
  6. plt.xlabel("ages")
  7. plt.ylabel("net worths")

unnamed.png