Я пытаюсь создать топор в качестве матрицы элементов (my_data), а y - в качестве вектора ответа (цель), используя python
import numpy as np
import pandas as pd
from sklearn.tree import DecisionTreeClassifier
my_data = pd.read_csv("Dataset.csv", delimiter=",")
my_data[0:5]
X = my_data[['Age', 'Gender', 'Psychological_Level', 'Social_Level']].values
X[0:5]
Над кодом выполняется правильно, но когда я пытаюсь запустить код нижевыдает ошибку "индекс 4 выходит за границы оси 1 с размером 4"
from sklearn import preprocessing
le_Gender = preprocessing.LabelEncoder()
le_Gender.fit(['Female','Male'])
X[:,1] = le_Gender.transform(X[:,1])
le_Psychological_Level = preprocessing.LabelEncoder()
le_Psychological_Level.fit([ 'likely to be well', 'mild mental disorder', 'severe mental disorder','moderate mental disorder'])
X[:,2] = le_Psychological_Level.transform(X[:,2])
le_Social_Level = preprocessing.LabelEncoder()
le_Social_Level.fit([ 'Moderately Mentally Healthy', 'Flourishing','Languishing'])
X[:,3] = le_Social_Level.transform(X[:,3])
le_Age = preprocessing.LabelEncoder()
le_Age.fit(['25','23','24','22 or less'])
X[:,4] = le_Age.transform(X[:,4])
X[0:5]