【发布时间】:2017-04-28 22:40:33
【问题描述】:
谁能简单地向我解释一下? 为了您的方便,我附上了完整的代码。
我有这段代码可以加载 IRIS 数据集并运行 SVM:
from sklearn import svm
import pandas as pd
def prepare_iris_DS():
print("Loading iris DS...")
url = 'http://archive.ics.uci.edu/ml/machine-learning-databases/iris/iris.data'
iris = pd.read_csv(url, names=["sepal length", "sepal width", "petal length", "petal width", "Species"])
df = pd.DataFrame(iris, columns=["sepal length", "sepal width", "petal length", "petal width", "Species"])
df.head()
iris.head()
print("Iris DS is Loaded")
columns, labels = ["sepal length", "sepal width"], ["Iris-setosa", "Iris-virginica"]
total = df.shape[0]
df = df[df.Species.isin(labels)]
X = df[columns]
print("selected {0} entries out of {1} from the dataset based on labels {2}".format(len(X), total, str(labels)))
Y = df[["Species"]]
Y.loc[Y.Species != labels[0], 'Species'] = 0.0
Y.loc[Y.Species == labels[0], 'Species'] = 1.0
X = X.as_matrix()
Y = Y.as_matrix()
return X, Y
X, Y = prepare_iris_DS()
rbf_svc = svm.SVC(kernel='rbf', gamma=0.1, C=0.1)
rbf_svc.fit(X, Y)
我在最后一行不断收到错误:rbf_svc.fit(X, Y)
File "C:\Anaconda2\lib\site-packages\sklearn\utils\multiclass.py", line 172, in check_classification_targets
raise ValueError("Unknown label type: %r" % y_type)
ValueError: Unknown label type: 'unknown'
但是...
当我输入这个命令时,它就可以正常工作了。
我不明白为什么?我很欣赏一个清晰/简单的答案
Y = Y.as_matrix().astype(float)
【问题讨论】:
标签: python pandas scikit-learn dataset