数据挖掘 决策树，文件报错

# clf = DecisionTreeClassifier() score:0.7836742214851667
classes = [' <=50K', ' >50K']
clf = DecisionTreeClassifier(criterion = 'entropy', max_depth = 8, min_samples_split = 5)
clf = clf.fit(train_dataset, train_target)
pred = clf.predict(test_dataset)
print(pred)
score = clf.score(test_dataset, test_target)
# pred = clf.predict_proba(test_dataset)
print(score)
# print(np.argmax(pred, axis = 1))

ValueError                                Traceback (most recent call last)
C:\Users\151354~1\AppData\Local\Temp/ipykernel_3344/854390936.py in <module>
      2 classes = [' <=50K', ' >50K']
      3 clf = DecisionTreeClassifier(criterion = 'entropy', max_depth = 8, min_samples_split = 5)
----> 4 clf = clf.fit(train_dataset, train_target)
      5 pred = clf.predict(test_dataset)
      6 print(pred)

机器学习实战：决策树的存储读写文件报错
http://t.zoukankan.com/huiAlex-p-8855810.html

   Traceback (most recent call last)
   值错误回溯（最近一次调用）
思路：
1、检查下函数使用，是否正确
2、看下是不是数据匹配出现了问题