python3运行随机森林代码时出现:DataConversionWarning

当我敲完随机森林的代码然后运行时出现如下balabala一堆东西:

D:\anaconda\envs\anaconda3\lib\site-packages\sklearn\model_selection\_validation.py:515:

DataConversionWarning: A column-vector y was passed when a 1d array was expected. Please change the shape of y to (n_samples,), for example using ravel().

estimator.fit(X_train, y_train, **fit_params)

后来发现是做标签的时候出问题了,原本做标签的代码是这样:

def Label(a,b):   
    label = []
    for i in range(a):
        if i1 < b:
            label.append(1)
        else:
            label.append(0)
    
    label = np.array(label).reshape(a,1)
    return (label)

我给样本做标签时多敲了一段代码,将这段代码直接zhus注释掉就行了

你可能感兴趣的:(机器学习)