【发布时间】:2021-07-28 20:46:09
【问题描述】:
您好,我是机器学习的新手,正在从事一个基于犯罪预测的有趣项目。我遇到了一个错误,现在已修复,但不幸的是,以下代码块返回了一个新错误。我正在使用UCI ML Repo 上提供的数据集。我检查了类似的帖子,但没有找到任何相关的解决方案。
import pandas as pd
import numpy as np
from sklearn import tree
from sklearn.model_selection import cross_val_score
df_d=pd.read_csv('communities-crime-full.csv')
df
df['highCrime'] = np.where(df['ViolentCrimesPerPop']>0.1, 1, 0)
Y = df['highCrime']
# print('total len is ',len(Y))
initial=pd.read_csv('communities-crime-full.csv')
initial = initial.drop('communityname', 1)
initial = initial.drop('ViolentCrimesPerPop', 1)
initial = initial.drop('fold', 1)
initial = initial.drop('state', 1)
initial = initial.drop('community', 1)
initial = initial.drop('county', 1)
skipinitialspace = True
feature_name=list(initial)
#initial=initial.convert_objects(convert_numeric=True)
initial = initial.apply(pd.to_numeric, errors='coerce')
New_data=initial.fillna(initial.mean())
# print('before...')
# print(initial)
# print('after...')
# print(New_data)
clf = tree.DecisionTreeClassifier(max_depth=3)
# clf = tree.DecisionTreeClassifier()
clf = clf.fit(New_data, Y)
clf
fold=df['fold']
scores = cross_val_score(clf, New_data, Y,fold,'accuracy',10)
print('cross_val_accuracy is ',scores)
print('cross_val_accuracy_avg is ',np.array(scores).mean())
scores = cross_val_score(clf, New_data, Y,fold,'precision',10)
print('cross_val_precision is ',scores)
print('cross_val_precision_avg is ',np.array(scores).mean())
scores = cross_val_score(clf, New_data, Y,fold,'recall',10)
print('cross_val_recall is ',scores)
print('cross_val_recall_avg is ',np.array(scores).mean())
错误:
ValueError Traceback (most recent call last)
<ipython-input-15-444381be2864> in <module>()
25 clf = tree.DecisionTreeClassifier(max_depth=3)
26 # clf = tree.DecisionTreeClassifier()
---> 27 clf = clf.fit(New_data, Y)
28 clf
29 fold=df['fold']
/root/.local/lib/python3.7/site-packages/sklearn/tree/_classes.py in fit(self, X, y, sample_weight, check_input, X_idx_sorted)
281 if len(y) != n_samples:
282 raise ValueError("Number of labels=%d does not match "
--> 283 "number of samples=%d" % (len(y), n_samples))
284 if not 0 <= self.min_weight_fraction_leaf <= 0.5:
285 raise ValueError("min_weight_fraction_leaf must in [0, 0.5]")
ValueError: Number of labels=1993 does not match number of samples=1994
【问题讨论】:
标签: python pandas machine-learning decision-tree valueerror