scikit-learn CART String Data

Question

Вы можете обучить DecisionTreeClassifier строковыми данными?

Когда я пытаюсь использовать данные Струнный я получаю ValueError: может не преобразователь строки плавать

clf = DecisionTreeClassifier()
clf.fit([['asdf', '1'], ['asdf', '0']], ['2', '3'])

Answer 1

Вы должны преобразовать строковые функции для числовых единиц в массиве NumPy; DictVectorizer делает это за вас.

samples = [['asdf', '1'], ['asdf', '0']] 
# turn the samples into dicts 
samples = [dict(enumerate(sample)) for sample in samples] 

# turn list of dicts into a numpy array 
vect = DictVectorizer(sparse=False) 
X = vect.fit_transform(samples) 

clf = DecisionTreeClassifier() 
clf.fit(X, ['2', '3']) 

Не забывайте использовать vect.transform на тестовых образцах, после преобразования тех dicts.