Как обучить древовидный классификатор с помощью строковых меток и функций? - PullRequest
0 голосов
/ 28 мая 2018

Ниже приведен мой код:

import sklearn
#features = [[140,"smooth"],[130,"smooth"],[150,"bumpy"],[170,"bumpy"]]
#labels = ["apple","apple","orange","orange"]

# Now replace 1 for smooth & 0 for bumpy and 0 for apple & 1 for orange
features = [[140,1],[130,1],[150,0],[170,0]]
labels = [0,0,1,1]

# Now I train a classifier
from sklearn import tree

my_classifier = tree.DecisionTreeClassifier()
my_classifier.fit(features,labels)
predict = my_classifier.predict([[150,0]])
print(predict)

Как я могу обучить классификатор без преобразования его в числа?

например, я хочу, чтобы ниже строки кода классифицировали мой классификатор.Пожалуйста, предложите, спасибо заранее:)

features = [[140,"smooth"],[130,"smooth"],[150,"bumpy"],[170,"bumpy"]]
labels = ["apple","apple","orange","orange"]

1 Ответ

0 голосов
/ 28 мая 2018

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

Вы можете просто сохранить словарь, индекс котороговектор представляет какую метку и повернуть их обратно после логического вывода.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...