python - 我正在尝试从 google youtube 视频重新创建一个简单的 helloworld 机器学习示例，但我遇到了我不理解的错误

Question

我正在尝试关注这个机器学习 hello world 视频：https ://www.youtube.com/watch?v=cKxRvEZd3Mw 但我遇到了我不明白的错误。错误似乎在库函数本身中，而不是我直接写的。我知道这个视频是 2016 年的，所以那段时间肯定发生了一些变化。我刚开始学习机器学习，需要帮助！

我检查了语法是一样的，不知道还能做什么

from sklearn import tree
features = [[140, 1], [130, 1], [150, 0], [[170, 0]]]
labels = [0, 0, 1, 1]
clf = tree.DecisionTreeClassifier()
clf = clf.fit(features, labels)
print(clf.predict([[150, 0]]))

视频中的输出是“[1]”，但我收到了这些错误：

C:\Users\offic\PycharmProjects\helloWorld\venv\Scripts\python.exe "C:/Users/offic/PycharmProjects/helloWorld/Hello World.py"
Traceback (most recent call last):
  File "C:/Users/offic/PycharmProjects/helloWorld/Hello World.py", line 5, in <module>
    clf = clf.fit(features, labels)
  File "C:\Users\offic\PycharmProjects\helloWorld\venv\lib\site-packages\sklearn\tree\tree.py", line 801, in fit
    X_idx_sorted=X_idx_sorted)
  File "C:\Users\offic\PycharmProjects\helloWorld\venv\lib\site-packages\sklearn\tree\tree.py", line 116, in fit
    X = check_array(X, dtype=DTYPE, accept_sparse="csc")
  File "C:\Users\offic\PycharmProjects\helloWorld\venv\lib\site-packages\sklearn\utils\validation.py", line 527, in check_array
    array = np.asarray(array, dtype=dtype, order=order)
  File "C:\Users\offic\PycharmProjects\helloWorld\venv\lib\site-packages\numpy\core\numeric.py", line 538, in asarray
    return array(a, dtype, copy=False, order=order)
ValueError: setting an array element with a sequence.

Process finished with exit code 1

标签： pythonpython-3.xtensorflowmachine-learningscikit-learn