web-dev-qa-db-ja.com

sklearn / pythonで「ValueError：期待される2D配列、代わりに1D配列を得た」の修正方法

私がいます。私は機械学習から始めて、簡単な例を試して学習しました。そこで、分類子を使用して、ファイルタイプに基づいてディスク内のファイルを分類したいと思います。私が書いたコードは、

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 <module>
  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.

誰かがコードで私を助けることができれば、それは私にとってとても役に立ちます！

pythonarraysnumpyscikit-learn
3
Karthik Bhojaraj
X=dataset.iloc[:, 0].values
y=dataset.iloc[:, 1].values

regressor=LinearRegression()
X=X.reshape(-1,1)
regressor.fit(X,y)

次のコードがありました。形状変更演算子はインプレース演算子ではありません。したがって、その値を上記のように再整形した後の値に置き換える必要があります。

0
Ameya Marathe