1. 程式人生 > 實用技巧 >機器學習02 Linear regression

機器學習02 Linear regression

機器學習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);

3: regressor.coef_
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