# 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)