機器學習02 Linear regression
阿新 • • 發佈:2020-10-09
機器學習100天 day02
Linear regression
注意:1、fit中輸入自變數和因變數,需要都是array型別,如果是Series需要將其進行轉換,X_train = np.array(X_train); trainX_train = X_train.reshape(len(X_train),1),轉換後會是shape會是(len(X_train),1)
regressor = linear_model.LinearRegression
1: regressor.fit(X_train,Y_train);
2: Y_pred = regressor.predict(Y_test);
4:注意plt.sactter(),plt.plot(X_test, regressor.predict(X_test))
參考連結
[1]: https://github.com/MLEveryday/100-Days-Of-ML-Code
[2]: https://github.com/MLEveryday/100-Days-Of-ML-Code/blob/master/Code/Day%202_Simple_Linear_Regression.md