在训练决策树模型时遇到错误

时间:2021-06-10 04:02:28

标签: python scikit-learn decision-tree

我正在尝试训练决策树模型,但我一直遇到错误,代码如下:

from sklearn.tree import DecisionTreeClassifier
dtree = DecisionTreeClassifier()
dtree.fit(X_train, y_train)

我已成功导入此代码所需的模块,但遇到错误:

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-91-b6417fbfb8de> in <module>
  1 from sklearn.tree import DecisionTreeClassifier
  2 dtree = DecisionTreeClassifier()
----> 3 dtree.fit(X_train, y_train)

~\Anaconda3\lib\site-packages\sklearn\tree\_classes.py in fit(self, X, y, sample_weight, 
check_input, X_idx_sorted)
888         """
889 
--> 890         super().fit(
891             X, y,
892             sample_weight=sample_weight,

~\Anaconda3\lib\site-packages\sklearn\tree\_classes.py in fit(self, X, y, sample_weight, 
check_input, X_idx_sorted)
154             check_X_params = dict(dtype=DTYPE, accept_sparse="csc")
155             check_y_params = dict(ensure_2d=False, dtype=None)
--> 156             X, y = self._validate_data(X, y,
157                                        validate_separately=(check_X_params,
158                                                             check_y_params))

~\Anaconda3\lib\site-packages\sklearn\base.py in _validate_data(self, X, y, reset, 
validate_separately, **check_params)
427                 # :(
428                 check_X_params, check_y_params = validate_separately
--> 429                 X = check_array(X, **check_X_params)
430                 y = check_array(y, **check_y_params)
431             else:

 ~\Anaconda3\lib\site-packages\sklearn\utils\validation.py in inner_f(*args, **kwargs)
 70                           FutureWarning)
 71         kwargs.update({k: arg for k, arg in zip(sig.parameters, args)})
 ---> 72         return f(**kwargs)
 73     return inner_f
 74 

 ~\Anaconda3\lib\site-packages\sklearn\utils\validation.py in check_array(array, 
 accept_sparse, accept_large_sparse, dtype, order, copy, force_all_finite, ensure_2d, 
 allow_nd, ensure_min_samples, ensure_min_features, estimator)
596                     array = array.astype(dtype, casting="unsafe", copy=False)
597                 else:
--> 598                     array = np.asarray(array, order=order, dtype=dtype)
599             except ComplexWarning:
600                 raise ValueError("Complex data not supported\n"

~\Anaconda3\lib\site-packages\numpy\core\_asarray.py in asarray(a, dtype, order)
 81 
 82     """
 ---> 83     return array(a, dtype, copy=False, order=order)
 84 
 85 

 ~\Anaconda3\lib\site-packages\pandas\core\generic.py in __array__(self, dtype)
 1779 
 1780     def __array__(self, dtype=None) -> np.ndarray:
 -> 1781         return np.asarray(self._values, dtype=dtype)
 1782 
 1783     def __array_wrap__(self, result, context=None):

 ~\Anaconda3\lib\site-packages\numpy\core\_asarray.py in asarray(a, dtype, order)
 81 
 82     """
 ---> 83     return array(a, dtype, copy=False, order=order)
 84 
 85 

 ValueError: could not convert string to float: 'SFC/CBU Reversal'

0 个答案:

没有答案