[機器學習]葡萄酒質量和年份關系

• 簡介
• 代碼

簡介

調用葡萄酒質量的csv文件，對葡萄酒質量與年份的關系進行探究

代碼

import numpy as np import matplotlib.pyplot as plt from sklearn.linear_model import LinearRegression from sklearn.model_selection import train_test_split data = np.genfromtxt('linear.csv',delimiter = ',') plt.scatter(data[1:,0],data[1:,1]) plt.title('Age VS Quality (test set)') plt.xlabel('Age') plt.ylabel('Quality') plt.show() x_train,x_test,y_train,y_test = train_test_split(data[1:,0],data[1:,1],test_size = 0.3) x_train = x_train[:,np.newaxis] #給數據增加維度，LinearRegression需要2維數據； 變成n行一列 x_test = x_test[:,np.newaxis] model = LinearRegression() model.fit(x_train,y_train) #訓練集的散點圖 plt.scatter(x_train,y_train, c = 'b') #模型對訓練集預測 plt.plot(x_train,model.predict(x_train),c = 'r',linewidth = 5) plt.title('Age VS Quality(Training)') plt.xlabel('Age') plt.ylabel('Quality') plt.show() #測試集的散點圖 plt.scatter(x_test,y_test, c = 'b' #模型對測試集預測 plt.plot(x_test,model.predict(x_test),c = 'r',linewidth = 5) plt.title('Age VS Quality(Testing)') plt.xlabel('Age') plt.ylabel('Quality') plt.show()

總結

以上是生活随笔為你收集整理的机器学习项目2-葡萄酒质量和年份的关系的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。