Как я могу исправить «ValueError: ожидался 2D-массив, вместо этого получен 1D-массив» в scikit-learn/Python? ⇐ Python

[Гость]

Я только начал изучать машинное обучение с простого примера, чтобы попробовать его изучить. Итак, я хочу классифицировать файлы на моем диске по типу файла, используя классификатор. Код, который я написал,

Код: Выделить всё

import sklearn
import numpy as np

# Importing a local data set from the desktop
import pandas as pd
mydata = pd.read_csv('file_format.csv',skipinitialspace=True)
print mydata

x_train = mydata.script
y_train = mydata.label

#print x_train
#print y_train
x_test = mydata.script

from sklearn import tree
classi = tree.DecisionTreeClassifier()

classi.fit(x_train, y_train)

predictions = classi.predict(x_test)
print predictions

И я получаю сообщение об ошибке:

Код: Выделить всё

  script  class  div   label
0       5      6    7    html
1       0      0    0  python
2       1      1    1     csv
Traceback (most recent call last):
File "newtest.py", line 21, in 
classi.fit(x_train, y_train)
File "/home/initiouser2/.local/lib/python2.7/site-
packages/sklearn/tree/tree.py", line 790, in fit
X_idx_sorted=X_idx_sorted)
File "/home/initiouser2/.local/lib/python2.7/site-
packages/sklearn/tree/tree.py", line 116, in fit
X = check_array(X, dtype=DTYPE, accept_sparse="csc")
File "/home/initiouser2/.local/lib/python2.7/site-
packages/sklearn/utils/validation.py", line 410, in check_array
"if it contains a single sample.".format(array))
ValueError: Expected 2D array, got 1D array instead:
array=[ 5.  0.  1.].
Reshape your data either using array.reshape(-1, 1) if your data has a
single feature or array.reshape(1, -1) if it contains a single sample.

Как решить эту проблему?


Подробнее здесь: https://stackoverflow.com/questions/466 ... -scikit-le