是否可以使用DecisionTreeRegressor計算Acuuracy?
2019/06/09 10:52 下午
機器學習共學討論版
Rosa
觀看數:13
回答數:3
收藏數:0
ml100-2
ml100-2-d47
如果是使用boston的資料集,所以會利用DecisionTreeRegressor 進行建模,但要如何計算ACC呢?
# 讀取波士頓地區的房屋價格資料集
boston = datasets.load_boston()
# 切分訓練集/測試集
x_train, x_test, y_train, y_test = train_test_split(boston.data, boston.target, test_size=0.25, random_state=4)
# 建立模型
Reg = DecisionTreeRegressor()
# 訓練模型
Reg.fit(x_train, y_train)
# 預測測試集
y_pred = Reg.predict(x_test)
acc = metrics.accuracy_score(y_test, y_pred)
print("Acuuracy: ", acc)
執行下面這一段落時就會發生錯誤
acc = metrics.accuracy_score(y_test, y_pred)
print("Acuuracy: ", acc)