har*_*shi 2 python machine-learning linear-regression scikit-learn
我是这个机器学习的新手,并使用这个波士顿数据集进行预测。除了precision_score 和accuracy_score 的结果外,一切都工作正常。这就是我所做的:
import pandas as pd
import sklearn
from sklearn.linear_model import LinearRegression
from sklearn import preprocessing,cross_validation, svm
from sklearn.datasets import load_boston
import numpy as np
from sklearn.metrics import accuracy_score, f1_score, precision_score, recall_score, classification_report, confusion_matrix
boston = load_boston()
df = pd.DataFrame(boston.data)
df.columns= boston.feature_names
df['Price']= boston.target
X = np.array(df.drop(['Price'],axis=1), dtype=np.float64)
X = preprocessing.scale(X)
y = np.array(df['Price'], dtype=np.float64)
print (len(X[:,6:7]),len(y))
X_train,X_test,y_train,y_test=cross_validation.train_test_split(X,y,test_size=0.30)
clf =LinearRegression()
clf.fit(X_train,y_train)
y_predict = clf.predict(X_test)
print(y_predict,len(y_predict))
print (accuracy_score(y_test, y_predict))
print(precision_score(y_test, y_predict,average = 'macro'))
Run Code Online (Sandbox Code Playgroud)
现在我收到以下错误:
文件“LinearRegression.py”,第 33 行,在
Run Code Online (Sandbox Code Playgroud)accuracy = accuracy_score(y_test, y_predict) File "/usr/local/lib/python2.7/dist-packages/sklearn/metrics/classification.py",第 172 行,在accuracy_score 中
Run Code Online (Sandbox Code Playgroud)y_type, y_true, y_pred = _check_targets(y_true, y_pred)文件“/usr/local/lib/python2.7/dist-packages/sklearn/metrics/classification.py”,第 89 行,在 _check_targets
Run Code Online (Sandbox Code Playgroud)raise ValueError("{0} is not supported".format(y_type)) ValueError: continuous is not supported
您正在使用线性回归模型作为
clf = LinearRegression()
Run Code Online (Sandbox Code Playgroud)
预测连续值。例如:1.2、1.3
而accuracy_score(y_test, y_predict)期望布尔值。1 或 0(真或假)或分类值,如 1、2、3、4 等。其中数字充当类别。
这就是您收到错误的原因。
如何解决这个问题?
由于您试图预测Price波士顿数据,这是一个连续值。我建议您将误差度量从准确度更改为 RMSE 或MSE
代替:
print(accuracy_score(y_test, y_predict))
Run Code Online (Sandbox Code Playgroud)
和:
from sklearn.metrics import mean_squared_error
print(mean_squared_error(y_test, y_predict))
Run Code Online (Sandbox Code Playgroud)
那将解决您的问题。